Home
last modified time | relevance | path

Searched +path:tensorflow +path:compiler +path:xla +path:service +path:spmd (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A DBUILD
H A Dspmd_partitioner.cc
H A Dgather_scatter_handler.cc
H A Dcustom_call_handler.h
H A Dstateful_rng_spmd_partitioner.h
H A Dspmd_partitioner_util.cc
H A Ddot_handler.cc
H A Dschedule_aware_collective_ops_cse.h
H A Dcanonicalize_all_gather_for_cse.cc
H A Dspmd_partitioner_util.h
H A Dcanonicalize_all_gather_for_cse.h
H A Dcanonicalize_all_gather_for_cse_test.cc
H A Dfft_handler.cc
H A Dcustom_call_handler.cc
H A Dconvolution_handler.cc
H A Dconvolution_handler.h
H A Dstateful_rng_spmd_partitioner.cc
H A Dschedule_aware_collective_ops_cse_test.cc
H A Dstateful_rng_spmd_partitioner_test.cc
H A Dspmd_partitioner_test.cc
H A Dspmd_partitioner.h
H A Dschedule_aware_collective_ops_cse.cc