Searched defs:input_feature_dimension (Results 1 – 5 of 5) sorted by relevance
79 int64 input_feature_dimension = dim_numbers.input_feature_dimension(); in ConvertBatchGroupedToFeatureGroupedConvolution() local554 int64 input_feature_dimension = dnums.kernel_input_feature_dimension(); in MatchBackwardInput() local
218 const int64 input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleBatchGroupCount() local584 const int64 input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleConvolution() local
519 int64 input_feature_dimension = 8; field
93 const int input_feature_dimension = in matchAndRewrite() local
233 int64 input_feature_dimension = dnums.input_feature_dimension(); in GenNewConvDNums() local