Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc886 auto zero_bcast = state_.b->AddInstruction( in ReplicatePartial() local
892 padded_target_shape, zero_bcast, hlo_, offsets)); in ReplicatePartial()
1073 auto zero_bcast = state_.b->AddInstruction( in Broadcast() local
1076 shape, HloOpcode::kSelect, is_src_core, hlo(), zero_bcast)); in Broadcast()