Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module.cc179 instruction->set_scatter(new_scatter); in ReplaceComputations()
Dhlo_instruction.h1578 void set_scatter(HloComputation* computation);
Dhlo_instructions.h1144 void set_scatter(HloComputation* computation) { in set_scatter() function
Dhlo_instruction.cc3496 void HloInstruction::set_scatter(HloComputation* computation) { in set_scatter() function in xla::HloInstruction
3497 return Cast<HloSelectAndScatterInstruction>(this)->set_scatter(computation); in set_scatter()