Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgpu_spmd_partitioner_test.cc39 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::gpu::__anonacb149e80111::GpuSpmdPartitionerTest
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dstateful_rng_spmd_partitioner_test.cc39 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::spmd::__anond29975700111::StatefulRngSpmdPartitionerTest
H A Dspmd_partitioner.cc4461 StatusOr<bool> SpmdPartitioner::PartitionComputation( in PartitionComputation() function in xla::spmd::SpmdPartitioner
H A Dspmd_partitioner_test.cc40 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::spmd::__anon2ed485a30111::SpmdPartitioningTest