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