Searched defs:lhs_ext2 (Results 1 – 2 of 2) sorted by relevance
70 const int lhs_ext2 = extent(extended_lhs_shape, 2); in BatchMatMul() local161 const int lhs_ext2 = extent(extended_lhs_shape, 2); in BatchMatMul() local315 const int lhs_ext2 = extent(extended_lhs_shape, 2); in BatchMatMul() local421 const int lhs_ext2 = extent(extended_lhs_shape, 2); in BatchMatMul() local
71 const int lhs_ext2 = batch_matmul::extent(extended_lhs_shape, 2); in BatchMatMul() local129 const int lhs_ext2 = batch_matmul::extent(extended_lhs_shape, 2); in BatchMatMul() local218 const int lhs_ext2 = batch_matmul::extent(extended_lhs_shape, 2); in BatchMatMul() local