Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc104 bool rhs_output_feature_dim_is_partitioned = in PartitionConvolutionWithBatchGroupCount() local
108 if (!lhs_batch_dim_is_partitioned && !rhs_output_feature_dim_is_partitioned) { in PartitionConvolutionWithBatchGroupCount()
195 bool rhs_output_feature_dim_is_partitioned = in PartitionConvolutionWithFeatureGroupCount() local
200 !rhs_output_feature_dim_is_partitioned) { in PartitionConvolutionWithFeatureGroupCount()