Searched refs:nested_loop_state_shape_ (Results 1 – 1 of 1) sorted by relevance
479 nested ? nested_loop_state_shape_ : loop_state_shape_; in BuildIndependentBodyComputation()533 0, nested_loop_state_shape_, "loop_state")); in BuildNestedBodyComputation()626 return BuildWhileInstructionWithCustomInit(nested_loop_state_shape_, in BuildWhileInstruction_InitPointsToNonDistinct()668 ShapeUtil::Equal(loop_state_shape, nested_loop_state_shape_); in BuildWhileInstructionWithCustomInit()690 Shape nested_loop_state_shape_ = ShapeUtil::MakeTupleShape( member in xla::__anon2db0cb110111::WhileCopyInsertionTest1133 BuildConditionComputation(nested_loop_state_shape_)); in TEST_F()