Searched refs:nested_loop_state_shape_ (Results 1 – 1 of 1) sorted by relevance
544 nested ? nested_loop_state_shape_ : loop_state_shape_; in BuildIndependentBodyComputation()598 0, nested_loop_state_shape_, "loop_state")); in BuildNestedBodyComputation()701 return BuildWhileInstructionWithCustomInit(nested_loop_state_shape_, in BuildWhileInstruction_InitPointsToAmbiguous()715 return BuildWhileInstructionWithCustomInit(nested_loop_state_shape_, in BuildWhileInstruction_InitPointsToNonDistinct()757 ShapeUtil::Equal(loop_state_shape, nested_loop_state_shape_); in BuildWhileInstructionWithCustomInit()779 Shape nested_loop_state_shape_ = ShapeUtil::MakeTupleShape( member in xla::__anon1fa5445b0111::WhileCopyInsertionTest1021 BuildConditionComputation(nested_loop_state_shape_)); in TEST_F()