Searched refs:expected_dnums (Results 1 – 1 of 1) sorted by relevance
1703 std::vector<int64> expected_dnums(num_dims); in InferConvolveShape() local1704 std::iota(expected_dnums.begin(), expected_dnums.end(), 0); in InferConvolveShape()1715 if (input_dnums != expected_dnums) { in InferConvolveShape()1721 if (window_dnums != expected_dnums) { in InferConvolveShape()1727 if (output_dnums != expected_dnums) { in InferConvolveShape()