Home
last modified time | relevance | path

Searched refs:input_feature_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc148 const int64 input_feature_size = in PartitionConvolutionWithFeatureGroupCount() local
152 if (input_feature_size != kernel_output_feature_size || in PartitionConvolutionWithFeatureGroupCount()
153 input_feature_size % original_hlo->feature_group_count() != 0) { in PartitionConvolutionWithFeatureGroupCount()