Searched defs:input_feature_dimension (Results 1 – 6 of 6) sorted by relevance
79 int64_t input_feature_dimension = dim_numbers.input_feature_dimension(); in ConvertBatchGroupedToFeatureGroupedConvolution() local581 int64_t input_feature_dimension = dnums.kernel_input_feature_dimension(); in MatchBackwardInput() local
218 const int64_t input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleBatchGroupCount() local585 const int64_t input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleConvolution() local
195 IntegerAttr input_feature_dimension = parsed_dims.second[IOFeature]; in parseConvolutionDimensions() local
525 int64 input_feature_dimension = 8; field
99 const int input_feature_dimension = in matchAndRewrite() local473 const int input_feature_dimension = in IsSupportedConvOp() local
238 int64_t input_feature_dimension = dnums.input_feature_dimension(); in GenNewConvDNums() local