Searched defs:rhs_ext2 (Results 1 – 2 of 2) sorted by relevance
73 const int rhs_ext2 = extent(extended_rhs_shape, 2); in BatchMatMul() local164 const int rhs_ext2 = extent(extended_rhs_shape, 2); in BatchMatMul() local318 const int rhs_ext2 = extent(extended_rhs_shape, 2); in BatchMatMul() local
74 const int rhs_ext2 = batch_matmul::extent(extended_rhs_shape, 2); in BatchMatMul() local132 const int rhs_ext2 = batch_matmul::extent(extended_rhs_shape, 2); in BatchMatMul() local221 const int rhs_ext2 = batch_matmul::extent(extended_rhs_shape, 2); in BatchMatMul() local