Searched refs:rhs_slice_shape (Results 1 – 1 of 1) sorted by relevance
1457 Shape rhs_slice_shape(rhs->shape()); in OptimizeDotOfConcatHelper() local1458 rhs_slice_shape.set_dimensions(rhs_contracting_dim, sub_k); in OptimizeDotOfConcatHelper()1470 rhs_slice_shape, rhs, /*start_indices=*/start_indices, in OptimizeDotOfConcatHelper()