Searched refs:batch_dimension_numbers (Results 1 – 1 of 1) sorted by relevance
263 std::vector<int64> batch_dimension_numbers; in BatchDot() local265 batch_dimension_numbers.reserve(ndims - 2); in BatchDot()267 batch_dimension_numbers.push_back(i); in BatchDot()269 std::vector<int64> x_config = batch_dimension_numbers; in BatchDot()272 std::vector<int64> y_config = batch_dimension_numbers; in BatchDot()275 std::vector<int64> output_config = batch_dimension_numbers; in BatchDot()