Searched defs:output_feature_dimension (Results 1 – 4 of 4) sorted by relevance
221 int64 output_feature_dimension = dim_numbers.output_feature_dimension(); in HandleBatchGroupCount() local627 const int64 output_feature_dimension = dim_numbers.output_feature_dimension(); in HandleConvolution() local
555 int64 output_feature_dimension = dnums.kernel_output_feature_dimension(); in MatchBackwardInput() local
542 int64 output_feature_dimension = 10; field
282 int64 output_feature_dimension = dnums.output_feature_dimension(); in GenNewConvDNums() local