Searched defs:last_warp_dim (Results 1 – 1 of 1) sorted by relevance
198 std::vector<int64> broadcasted_dims, int64 last_warp_dim, in BoundSamples()251 int64 last_warp_dim, int64 data_channels, in CalculateGradData()375 const int64 last_warp_dim = warp_shape.dims() - 1; in CalculateGradWarp() local504 const int64 last_warp_dim = warp_shape.dims() - 1; in Compile() local626 const int64 last_warp_dim = warp_shape.dims() - 1; in Compile() local