Searched defs:dim_a (Results 1 – 5 of 5) sorted by relevance
53 void SwapDimSizes(const int dim_a, const int dim_b, TensorShape* shape) { in SwapDimSizes()
229 const int dim_a = i - rank_offset_a; in GetBroadcastLayoutForElementWise() local
673 size_t dim_a, in tensors_have_same_size_at_dims()
2059 int dim_a = 0; in TEST() local2208 int dim_a = 1; in TEST() local2239 int dim_a = 0; in TEST() local
388 auto static_dims = [](int64_t dim_a, int64_t dim_b) { in verify()