Searched refs:shape_pb (Results 1 – 2 of 2) sorted by relevance
265 std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb) { in ShapeProtoToVec() argument267 if (shape_pb.dim_size() == 0 && !shape_pb.unknown_rank()) { in ShapeProtoToVec()271 for (const auto& d : shape_pb.dim()) { in ShapeProtoToVec()279 TensorShapeProto shape_pb; in VecToShapeProto() local281 shape_pb.set_unknown_rank(true); in VecToShapeProto()282 return shape_pb; in VecToShapeProto()285 shape_pb.add_dim()->set_size(s); in VecToShapeProto()287 return shape_pb; in VecToShapeProto()
39 std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb);