Searched refs:mutable_tuple_shapes (Results 1 – 13 of 13) sorted by relevance
148 for (auto& element_shape : *shape->mutable_tuple_shapes()) { in SetToDefaultLayout()273 for (auto& element_shape : *shape->mutable_tuple_shapes()) { in ClearLayout()441 dst->mutable_tuple_shapes(i))); in CopyLayoutInternal()
146 Shape* mutable_tuple_shapes(int index) { return &tuple_shapes_.at(index); } in mutable_tuple_shapes() function153 std::vector<Shape>* mutable_tuple_shapes() { return &tuple_shapes_; } in mutable_tuple_shapes() function
143 result.mutable_tuple_shapes()->reserve(shapes.size()); in MakeTupleShapeImpl()442 *tuple_shape->mutable_tuple_shapes(index) = shape; in UpdateTupleShape()455 UpdateDynamicDimension(shape->mutable_tuple_shapes(index.front()), in UpdateDynamicDimension()1018 return_shape = return_shape->mutable_tuple_shapes(i); in GetMutableSubshape()1089 shape->mutable_tuple_shapes(i), func, index)); in ForEachMutableSubshapeHelper()
469 *shape.mutable_tuple_shapes(1)->mutable_tuple_shapes(0)->mutable_layout() = in TEST_F()
74 proto.mutable_tuple_shapes()->Reserve(tuple_shapes_size()); in ToProto()
172 new_shape.mutable_tuple_shapes(0)->set_element_type(F32); in FuseConvertToFloat()530 new_shape.mutable_tuple_shapes(0)->set_element_type(F16); in FuseConvertToF16()611 new_shape.mutable_tuple_shapes(0)->set_element_type(conv_output_ty); in FuseConvertToS8()
2523 *root->mutable_shape()->mutable_tuple_shapes(i) = output_shapes.at(i); in GetOrCreateSubComputationFromRegion()
737 *param->mutable_shape()->mutable_tuple_shapes(i) = param_gtes[i]->shape(); in RewriteLoopWithConcatGroups()738 *body->root_instruction()->mutable_shape()->mutable_tuple_shapes(i) = in RewriteLoopWithConcatGroups()740 *cond_param->mutable_shape()->mutable_tuple_shapes(i) = in RewriteLoopWithConcatGroups()742 *loop->mutable_shape()->mutable_tuple_shapes(i) = param_gtes[i]->shape(); in RewriteLoopWithConcatGroups()
2098 *hlo->mutable_shape()->mutable_tuple_shapes(i) = hlo->operand(i)->shape(); in HandleTuple()
1085 result.mutable_tuple_shapes()->reserve(operand_shapes.size()); in InferVariadicOpShape()
3013 result.mutable_tuple_shapes()->reserve(types.size()); in to_apply()
1831 auto subshapes = shape->mutable_tuple_shapes(); in ConvertLayout()1924 auto subshapes = shape->mutable_tuple_shapes(); in ConvertInfeedtLayout()1930 auto data_subshapes = datashape->mutable_tuple_shapes(); in ConvertInfeedtLayout()
453 param_shape_ptr = param_shape_ptr->mutable_tuple_shapes(index); in SetDynamicBinding()530 remove_dynamic_dimension(shape->mutable_tuple_shapes(i)); in Build()3169 *(shape.mutable_tuple_shapes(i)->mutable_layout()) = *layout; in AllToAllTuple()