Home
last modified time | relevance | path

Searched defs:spmd (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dstateful_rng_spmd_partitioner.cc24 namespace spmd { namespace
Dstateful_rng_spmd_partitioner_test.cc34 namespace spmd { namespace
Dcanonicalize_all_gather_for_cse_test.cc28 namespace spmd { namespace
Dschedule_aware_collective_ops_cse_test.cc28 namespace spmd { namespace
Dcustom_call_handler.cc41 namespace spmd { namespace
Dfft_handler.cc42 namespace spmd { namespace
Dconvolution_handler.cc36 namespace spmd { namespace
Dgather_scatter_handler.cc35 namespace spmd { namespace
Dspmd_partitioner_util.cc44 namespace spmd { namespace
Ddot_handler.cc45 namespace spmd { namespace
Dspmd_partitioner.cc61 namespace spmd { namespace
Dspmd_partitioner_test.cc31 namespace spmd { namespace