Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc475 int64 output_batch_partitions, int64 lhs_contracting_partitions, in PartitionBaseCase() argument
540 output_batch_partitions == num_partitions && in PartitionBaseCase()
557 output_batch_partitions == num_partitions && in PartitionBaseCase()
1550 if (output_batch_partitions == num_partitions) { in PartitionBaseCase()
2079 int64 output_batch_partitions, int64 output_lhs_non_contracting_partitions, in PartitionDotGroupOnContracting() argument
2196 output_batch_partitions == group_count) { in PartitionDotGroupOnContracting()
2476 const int64 output_batch_partitions = in PartitionDot() local
2506 output_batch_partitions > 1 || in PartitionDot()
2590 lhs_batch_partitions, rhs_batch_partitions, output_batch_partitions, in PartitionDot()
2604 if ((lhs_batch_partitions == output_batch_partitions || in PartitionDot()
[all …]