Searched refs:mutable_tuple_shapes (Results 1 – 7 of 7) sorted by relevance
114 Shape* mutable_tuple_shapes(int index) { return &tuple_shapes_.at(index); } in mutable_tuple_shapes() function121 std::vector<Shape>* mutable_tuple_shapes() { return &tuple_shapes_; } in mutable_tuple_shapes() function
145 for (auto& element_shape : *shape->mutable_tuple_shapes()) { in SetToDefaultLayout()261 for (auto& element_shape : *shape->mutable_tuple_shapes()) { in ClearLayout()389 dst->mutable_tuple_shapes(i))); in CopyLayoutInternal()
68 proto.mutable_tuple_shapes()->Reserve(tuple_shapes_size()); in ToProto()
471 *shape.mutable_tuple_shapes(1)->mutable_tuple_shapes(0)->mutable_layout() = in TEST_F()
240 result.mutable_tuple_shapes()->reserve(shapes.size()); in MakeTupleShape()812 return_shape = return_shape->mutable_tuple_shapes(i); in GetMutableSubshape()883 shape->mutable_tuple_shapes(i), func, index)); in ForEachMutableSubshapeHelper()
268 param_shape_ptr = param_shape_ptr->mutable_tuple_shapes(index); in SetDynamicBinding()336 remove_dynamic_dimension(shape->mutable_tuple_shapes(i)); in Build()
1039 result.mutable_tuple_shapes()->reserve(operand_shapes.size()); in InferVariadicOpShape()