Searched refs:kConvFeatureDimension (Results 1 – 2 of 2) sorted by relevance
2649 dimension_numbers.set_input_feature_dimension(kConvFeatureDimension); in CreateDefaultConvDimensionNumbers()2651 dimension_numbers.set_output_feature_dimension(kConvFeatureDimension); in CreateDefaultConvDimensionNumbers()
177 static constexpr int64 kConvFeatureDimension = 1; variable