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()
513 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildDependentBodyOneReadOnlyComputation()
544 nested ? nested_loop_state_shape_ : loop_state_shape_; in BuildIndependentBodyComputation()
659 loop_state_shape_, condition, body, loop_state_init)); in BuildWhileInstruction()
669 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToConstant()
677 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToParameter()
732 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering()
775 Shape loop_state_shape_ = member in xla::__anon1fa5445b0111::WhileCopyInsertionTest
795 BuildConditionComputation(loop_state_shape_)); in TEST_F()
824 BuildConditionComputation(loop_state_shape_)); in TEST_F()
[all …]
Dbuffer_assignment_test.cc1862 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildWhileConditionComputation()
1876 HloInstruction::CreateParameter(0, loop_state_shape_, "loop_state")); in BuildWhileBodyComputation()
1906 Shape loop_state_shape_ = member in xla::__anon3198ac380111::WhileBufferAssignmentTest
1941 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
1952 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, tuple1)); in TEST_F()
2260 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2268 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, while0)); in TEST_F()
2408 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple0)); in TEST_F()
2410 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple1)); in TEST_F()
2480 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
[all …]