Searched refs:output_dnums (Results 1 – 1 of 1) sorted by relevance
1656 std::vector<int64> output_dnums(num_dims); in InferConvolveShape() local1657 output_dnums[0] = dnums.output_batch_dimension(); in InferConvolveShape()1658 output_dnums[1] = dnums.output_feature_dimension(); in InferConvolveShape()1660 dnums.output_spatial_dimensions().end(), output_dnums.begin() + 2); in InferConvolveShape()1661 absl::c_sort(output_dnums); in InferConvolveShape()1669 !absl::c_all_of(output_dnums, in_range)) { in InferConvolveShape()1687 if (output_dnums != expected_dnums) { in InferConvolveShape()