Searched defs:input_feature_dimension (Results 1 – 5 of 5) sorted by relevance
502 int64_t input_feature_dimension = dnums.kernel_input_feature_dimension(); in MatchBackwardInput() local615 int64_t input_feature_dimension = dim_numbers.input_feature_dimension(); in ConvertBatchGroupedToFeatureGroupedConvolution() local
223 const int64_t input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleBatchGroupCount() local590 const int64_t input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleConvolution() local
213 const int64_t input_feature_dimension = dnums.getInputFeatureDimension(); in matchAndRewrite() local427 const int input_feature_dimension = dnums.getInputFeatureDimension(); in matchAndRewrite() local815 const int input_feature_dimension = dnums.getInputFeatureDimension(); in IsSupportedConvOp() local
567 int64 input_feature_dimension = 8; field
247 int64_t input_feature_dimension = dnums.input_feature_dimension(); in GenNewConvDNums() local