Searched defs:output_feature_dimension (Results 1 – 5 of 5) sorted by relevance
221 int64_t output_feature_dimension = dim_numbers.output_feature_dimension(); in HandleBatchGroupCount() local628 const int64_t output_feature_dimension = in HandleConvolution() local
211 IntegerAttr output_feature_dimension = parsed_dims.second[IOFeature]; in parseConvolutionDimensions() local
582 int64_t output_feature_dimension = dnums.kernel_output_feature_dimension(); in MatchBackwardInput() local
548 int64 output_feature_dimension = 10; field
288 int64_t output_feature_dimension = dnums.output_feature_dimension(); in GenNewConvDNums() local