Searched defs:PartitionComputation (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ | ||
| D | gpu_spmd_partitioner_test.cc | 39 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::gpu::__anon3531115e0111::GpuSpmdPartitionerTest |
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ | ||
| D | spmd_partitioner.cc | 3672 StatusOr<bool> SpmdPartitioner::PartitionComputation( in PartitionComputation() function in xla::spmd::SpmdPartitioner |
| D | spmd_partitioner_test.cc | 39 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::spmd::__anon16ebe0590111::SpmdPartitioningTest |