Searched refs:expected_dnums (Results 1 – 1 of 1) sorted by relevance
1702 std::vector<int64_t> expected_dnums(num_dims); in InferConvolveShape() local1703 std::iota(expected_dnums.begin(), expected_dnums.end(), 0); in InferConvolveShape()1716 if (input_dnums != expected_dnums) { in InferConvolveShape()1722 if (window_dnums != expected_dnums) { in InferConvolveShape()1728 if (output_dnums != expected_dnums) { in InferConvolveShape()