Searched refs:MakePartitioningState (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1402 PartitionedHlo(clone, hlo->shape(), MakePartitioningState()) in DefaultAction()
1560 MakePartitioningState(), grouped.device_groups, &b_); in HandleConcatenate()
1709 MakePartitioningState()); in HandleSort()
1719 MakePartitioningState()); in HandleSort()
1737 MakePartitioningState()); in HandleSort()
1852 MakePartitioningState()); in HandleCustomCall()
1863 MakePartitioningState()); in HandleCustomCall()
1890 MakePartitioningState()); in HandleCustomCall()
1918 MakePartitioningState()); in HandleCustomCall()
1940 MakePartitioningState()) in HandleCustomCall()
[all …]
|
| D | spmd_partitioner.h | 476 hlo, PartitionedHlo(new_hlo, hlo->shape(), MakePartitioningState())); in SetPartitionedHlo()
482 PartitionedHlo::PartitioningState MakePartitioningState() { in MakePartitioningState() function
|
| D | gather_scatter_handler.cc | 610 return PartitionedHlo(all_reduce, hlo->shape(), MakePartitioningState()) in HandleScatter()
629 return PartitionedHlo(pscatter, hlo->shape(), MakePartitioningState()) in HandleScatter()
661 return PartitionedHlo(pscatter, hlo->shape(), MakePartitioningState()) in HandleScatter()
|