Home
last modified time | relevance | path

Searched refs:update_sharding (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h164 const HloSharding& update_sharding, const HloInstruction& hlo);
Dhlo_sharding_util.cc1008 const HloSharding& update_sharding, const HloInstruction& hlo) { in ScatterOutputShardingFromUpdate() argument
1027 hlo.shape(), update_sharding, inserted_window_dims, in ScatterOutputShardingFromUpdate()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2364 auto update_sharding = HloSharding::Replicate(); in HandleDynamicUpdateSlice() local
2367 update_sharding = in HandleDynamicUpdateSlice()
2372 GetPartitionedHlo(hlo->operand(1)).Reshard(update_sharding).hlo(); in HandleDynamicUpdateSlice()