Searched refs:kernel_output_feature_size (Results 1 – 1 of 1) sorted by relevance
59 const int64 kernel_output_feature_size = in PartitionConvolutionWithBatchGroupCount() local61 if (input_batch_size != kernel_output_feature_size || in PartitionConvolutionWithBatchGroupCount()150 const int64 kernel_output_feature_size = in PartitionConvolutionWithFeatureGroupCount() local152 if (input_feature_size != kernel_output_feature_size || in PartitionConvolutionWithFeatureGroupCount()