Searched defs:PartitionComputation (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ | ||
| D | spmd_partitioner.cc | 3632 StatusOr<bool> SpmdPartitioner::PartitionComputation( in PartitionComputation() function in xla::spmd::SpmdPartitioner |
| D | spmd_partitioner_test.cc | 38 StatusOr<std::unique_ptr<HloModule>> PartitionComputation( in PartitionComputation() function in xla::spmd::__anonccb49f740111::SpmdPartitioningTest |