Searched refs:loop_state_shape_components (Results 1 – 1 of 1) sorted by relevance
230 std::vector<Shape> loop_state_shape_components; in MakeLoopStateShapeWithLayout() local231 loop_state_shape_components.reserve(init_values.size() + 1); in MakeLoopStateShapeWithLayout()232 loop_state_shape_components.push_back(ShapeUtil::MakeShape(S32, {})); in MakeLoopStateShapeWithLayout()234 std::back_inserter(loop_state_shape_components), in MakeLoopStateShapeWithLayout()242 return ShapeUtil::MakeTupleShape(loop_state_shape_components); in MakeLoopStateShapeWithLayout()