| copy_insertion_test.cc | 440 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state"));
513 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state"));
544 nested ? nested_loop_state_shape_ : loop_state_shape_;
659 loop_state_shape_, condition, body, loop_state_init));
669 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init,
677 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init,
732 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_,
775 Shape loop_state_shape_ = member in class:xla::__anon44322::WhileCopyInsertionTest
[all...] |
