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() local
69 const int lhs_ext1 = batch_matmul::extent(extended_lhs_shape, 1); in BatchMatMul() local127 const int lhs_ext1 = batch_matmul::extent(extended_lhs_shape, 1); in BatchMatMul() local216 const int lhs_ext1 = batch_matmul::extent(extended_lhs_shape, 1); in BatchMatMul() local