Searched refs:mutable_tuple_shapes (Results 1 – 11 of 11) sorted by relevance
147 for (auto& element_shape : *shape->mutable_tuple_shapes()) { in SetToDefaultLayout()263 for (auto& element_shape : *shape->mutable_tuple_shapes()) { in ClearLayout()379 dst->mutable_tuple_shapes(i))); in CopyLayoutInternal()
130 Shape* mutable_tuple_shapes(int index) { return &tuple_shapes_.at(index); } in mutable_tuple_shapes() function137 std::vector<Shape>* mutable_tuple_shapes() { return &tuple_shapes_; } in mutable_tuple_shapes() function
68 proto.mutable_tuple_shapes()->Reserve(tuple_shapes_size()); in ToProto()
368 result.mutable_tuple_shapes()->reserve(shapes.size()); in MakeTupleShape()399 *tuple_shape->mutable_tuple_shapes(index) = shape; in UpdateTupleShape()412 UpdateDynamicDimension(shape->mutable_tuple_shapes(index.front()), in UpdateDynamicDimension()939 return_shape = return_shape->mutable_tuple_shapes(i); in GetMutableSubshape()1010 shape->mutable_tuple_shapes(i), func, index)); in ForEachMutableSubshapeHelper()
423 *shape.mutable_tuple_shapes(1)->mutable_tuple_shapes(0)->mutable_layout() = in TEST_F()
346 conv->mutable_shape()->mutable_tuple_shapes(0)->set_element_type( in RewriteForConvertOrClampImpl()
2812 *root->mutable_shape()->mutable_tuple_shapes(i) = output_shapes.at(i); in GetOrCreateSubComputationFromRegion()
390 param_shape_ptr = param_shape_ptr->mutable_tuple_shapes(index); in SetDynamicBinding()467 remove_dynamic_dimension(shape->mutable_tuple_shapes(i)); in Build()2973 *(shape.mutable_tuple_shapes(i)->mutable_layout()) = *layout; in AllToAllTuple()
1750 *hlo->mutable_shape()->mutable_tuple_shapes(i) = hlo->operand(i)->shape(); in HandleTuple()
1083 result.mutable_tuple_shapes()->reserve(operand_shapes.size()); in InferVariadicOpShape()
1166 auto subshapes = shape->mutable_tuple_shapes(); in ConvertLayout()