Searched defs:lhs_ext1 (Results 1 – 2 of 2) sorted by relevance
69 const int lhs_ext1 = extent(extended_lhs_shape, 1); in BatchMatMul() local160 const int lhs_ext1 = extent(extended_lhs_shape, 1); in BatchMatMul() local314 const int lhs_ext1 = extent(extended_lhs_shape, 1); in BatchMatMul() local420 const int lhs_ext1 = extent(extended_lhs_shape, 1); in BatchMatMul() local
70 const int lhs_ext1 = batch_matmul::extent(extended_lhs_shape, 1); in BatchMatMul() local128 const int lhs_ext1 = batch_matmul::extent(extended_lhs_shape, 1); in BatchMatMul() local217 const int lhs_ext1 = batch_matmul::extent(extended_lhs_shape, 1); in BatchMatMul() local