Searched refs:dim_proto (Results 1 – 2 of 2) sorted by relevance
97 for (const auto& dim_proto : shape_pb_in.dim()) { in ReplaceUnknownShapeDim() local99 dim_proto.size() == -1) { in ReplaceUnknownShapeDim()104 dims.push_back(std::max<int32>(1, dim_proto.size())); in ReplaceUnknownShapeDim()105 shape_pb_out->add_dim()->set_size(dim_proto.size()); in ReplaceUnknownShapeDim()
543 TensorShapeProto_Dim dim_proto = TensorShapeProto_Dim(); in MakeTensorShapeFromDimensions() local544 dim_proto.set_size(dim); in MakeTensorShapeFromDimensions()545 *shape_proto.add_dim() = std::move(dim_proto); in MakeTensorShapeFromDimensions()