Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc101 bool lhs_batch_dim_is_partitioned = in PartitionConvolutionWithBatchGroupCount() local
108 if (!lhs_batch_dim_is_partitioned && !rhs_output_feature_dim_is_partitioned) { in PartitionConvolutionWithBatchGroupCount()
113 if (lhs_batch_dim_is_partitioned) { in PartitionConvolutionWithBatchGroupCount()