Searched defs:PartitionComputation (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ | ||
| D | stateful_rng_spmd_partitioner_test.cc | 39 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::spmd::__anon3b79eff00111::StatefulRngSpmdPartitionerTest |
| D | spmd_partitioner.cc | 4461 StatusOr<bool> SpmdPartitioner::PartitionComputation( in PartitionComputation() function in xla::spmd::SpmdPartitioner |
| D | spmd_partitioner_test.cc | 40 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::spmd::__anondc5bf0230111::SpmdPartitioningTest |
| /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::__anone77884680111::GpuSpmdPartitionerTest |