Searched refs:input_dnums (Results 1 – 1 of 1) sorted by relevance
1642 std::vector<int64> input_dnums(num_dims); in InferConvolveShape() local1643 input_dnums[0] = dnums.input_batch_dimension(); in InferConvolveShape()1644 input_dnums[1] = dnums.input_feature_dimension(); in InferConvolveShape()1646 dnums.input_spatial_dimensions().end(), input_dnums.begin() + 2); in InferConvolveShape()1647 absl::c_sort(input_dnums); in InferConvolveShape()1667 if (!absl::c_all_of(input_dnums, in_range) || in InferConvolveShape()1675 if (input_dnums != expected_dnums) { in InferConvolveShape()