Searched refs:window_dnums (Results 1 – 1 of 1) sorted by relevance
1689 std::vector<int64> window_dnums(num_dims); in InferConvolveShape() local1690 window_dnums[0] = dnums.kernel_input_feature_dimension(); in InferConvolveShape()1691 window_dnums[1] = dnums.kernel_output_feature_dimension(); in InferConvolveShape()1693 dnums.kernel_spatial_dimensions().end(), window_dnums.begin() + 2); in InferConvolveShape()1694 absl::c_sort(window_dnums); in InferConvolveShape()1708 !absl::c_all_of(window_dnums, in_range) || in InferConvolveShape()1721 if (window_dnums != expected_dnums) { in InferConvolveShape()