Searched refs:batch_dim_sizes (Results 1 – 1 of 1) sorted by relevance
181 std::vector<int64> batch_dim_sizes; in CanonicalizeDot() local182 batch_dim_sizes.reserve(num_batch_dims); in CanonicalizeDot()188 batch_dim_sizes.push_back(lhs_shape.dimensions(i)); in CanonicalizeDot()211 std::vector<int64> lhs_reshape_dims = batch_dim_sizes; in CanonicalizeDot()256 std::vector<int64> rhs_reshape_dims = batch_dim_sizes; in CanonicalizeDot()265 std::vector<int64> dot_dims = batch_dim_sizes; in CanonicalizeDot()