Searched refs:kernel_output_feature_size (Results 1 – 1 of 1) sorted by relevance
59 const int64_t kernel_output_feature_size = in PartitionConvolutionWithBatchGroupCount() local61 if (input_batch_size != kernel_output_feature_size || in PartitionConvolutionWithBatchGroupCount()150 const int64_t kernel_output_feature_size = in PartitionConvolutionWithFeatureGroupCount() local152 if (kernel_output_feature_size % original_hlo->feature_group_count() != 0 || in PartitionConvolutionWithFeatureGroupCount()