Searched refs:lhs_reshape_dims (Results 1 – 1 of 1) sorted by relevance
211 std::vector<int64> lhs_reshape_dims = batch_dim_sizes; in CanonicalizeDot() local212 lhs_reshape_dims.push_back(lhs_non_contracting_size); in CanonicalizeDot()213 lhs_reshape_dims.push_back(lhs_contracting_size); in CanonicalizeDot()217 ShapeUtil::MakeShape(lhs_shape.element_type(), lhs_reshape_dims), in CanonicalizeDot()