Searched refs:loop_state_init (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer_test.cc | 96 auto loop_state_init = in BuildWhileInstruction() local
104 condition, body, loop_state_init)); in BuildWhileInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 648 auto loop_state_init = builder.AddInstruction( in BuildWhileInstruction() local
651 loop_state_init->shape(), condition, body, loop_state_init)); in BuildWhileInstruction()
656 auto loop_state_init = builder.AddInstruction( in BuildWhileInstruction() local
659 loop_state_shape_, condition, body, loop_state_init)); in BuildWhileInstruction()
764 auto loop_state_init = builder->AddInstruction( in BuildWhileInstructionWithCustomInit() local
767 loop_state_shape, condition, body, loop_state_init)); in BuildWhileInstructionWithCustomInit()
|