Home
last modified time | relevance | path

Searched defs:replicated_sharding (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dcustom_call_handler.cc107 auto replicated_sharding = HloSharding::Replicate(); in HandleCustomCallTopK() local
H A Dspmd_partitioner.cc1719 auto replicated_sharding = in PatternMatchPartiallyReplicateDim() local
3057 HloSharding replicated_sharding = in HandleDynamicUpdateSlice() local