Home
last modified time | relevance | path

Searched defs:PartitionComputation (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_spmd_partitioner_test.cc39 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::gpu::__anon3531115e0111::GpuSpmdPartitionerTest
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3672 StatusOr<bool> SpmdPartitioner::PartitionComputation( in PartitionComputation() function in xla::spmd::SpmdPartitioner
Dspmd_partitioner_test.cc39 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::spmd::__anon16ebe0590111::SpmdPartitioningTest