Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcopy_insertion_test.cc440 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildDependentBodyComputation()
516 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildDependentBodyOneReadOnlyComputation()
550 nested ? nested_loop_state_shape_ : loop_state_shape_; in BuildIndependentBodyComputation()
665 loop_state_shape_, condition, body, loop_state_init)); in BuildWhileInstruction()
675 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToConstant()
683 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToParameter()
738 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering()
781 Shape loop_state_shape_ = member in xla::__anon1fa5445b0111::WhileCopyInsertionTest
801 BuildConditionComputation(loop_state_shape_)); in TEST_F()
1029 BuildConditionComputation(loop_state_shape_)); in TEST_F()
[all …]
Dbuffer_assignment_test.cc2011 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildWhileConditionComputation()
2025 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildWhileBodyComputation()
2054 Shape loop_state_shape_ = member in xla::__anon3198ac380111::WhileBufferAssignmentTest
2089 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2100 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, tuple1)); in TEST_F()
2407 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2415 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, while0)); in TEST_F()
2590 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple0)); in TEST_F()
2592 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple1)); in TEST_F()
2661 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
[all …]