Searched defs:dim_b (Results 1 – 8 of 8) sorted by relevance
40 size_t dim_b, in upsample_nearest1d_out_frame()72 size_t dim_b, in upsample_nearest1d_backward_out_frame()
45 size_t dim_b, in upsample_nearest3d_out_frame()90 size_t dim_b, in upsample_nearest3d_backward_out_frame()
116 size_t dim_b, in upsample_nearest2d_backward_out_frame()
53 void SwapDimSizes(const int dim_a, const int dim_b, TensorShape* shape) { in SwapDimSizes()
230 const int dim_b = i - rank_offset_b; in GetBroadcastLayoutForElementWise() local
675 size_t dim_b) { in tensors_have_same_size_at_dims()
2060 int dim_b = 1; in TEST() local2215 int dim_b = 2; in TEST() local2245 int dim_b = 1; in TEST() local
388 auto static_dims = [](int64_t dim_a, int64_t dim_b) { in verify()