Searched defs:lhs_ext0 (Results 1 – 2 of 2) sorted by relevance
68 const int lhs_ext0 = extent(extended_lhs_shape, 0); in BatchMatMul() local159 const int lhs_ext0 = extent(extended_lhs_shape, 0); in BatchMatMul() local313 const int lhs_ext0 = extent(extended_lhs_shape, 0); in BatchMatMul() local419 const int lhs_ext0 = extent(extended_lhs_shape, 0); in BatchMatMul() local
69 const int lhs_ext0 = batch_matmul::extent(extended_lhs_shape, 0); in BatchMatMul() local127 const int lhs_ext0 = batch_matmul::extent(extended_lhs_shape, 0); in BatchMatMul() local216 const int lhs_ext0 = batch_matmul::extent(extended_lhs_shape, 0); in BatchMatMul() local