Searched refs:expected_dnums (Results 1 – 1 of 1) sorted by relevance
1663 std::vector<int64> expected_dnums(num_dims); in InferConvolveShape() local1664 std::iota(expected_dnums.begin(), expected_dnums.end(), 0); in InferConvolveShape()1675 if (input_dnums != expected_dnums) { in InferConvolveShape()1681 if (window_dnums != expected_dnums) { in InferConvolveShape()1687 if (output_dnums != expected_dnums) { in InferConvolveShape()