Searched defs:dynamic_update_slice (Results 1 – 11 of 11) sorted by relevance
51 std::optional<HloInstruction*> dynamic_update_slice; member208 std::optional<HloInstruction*> dynamic_update_slice; in IsAllReduceMovable() member433 const HloInstruction& dynamic_update_slice) -> bool { in IsAllReduceMovable()448 const HloInstruction& dynamic_update_slice) -> bool { in IsAllReduceMovable()
187 HloInstructionPtr dynamic_update_slice) override { in HandleDynamicUpdateSlice()
2116 auto dynamic_update_slice = in TEST_F() local2160 auto dynamic_update_slice = in TEST_F() local2409 auto dynamic_update_slice = in TEST_F() local2451 auto dynamic_update_slice = in TEST_F() local
912 auto dynamic_update_slice = builder.AddInstruction( in TEST_F() local
270 const HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice()
1225 HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice()
5039 HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice()
57 bool CanUpdateDynamicSliceInPlace(HloInstruction* dynamic_update_slice, in CanUpdateDynamicSliceInPlace()197 auto* dynamic_update_slice = fusion->root_instruction(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
32 namespace dynamic_update_slice { namespace
367 dynamic_update_slice = gen_xla_ops.xla_dynamic_update_slice variable
2089 HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice()