Searched refs:shape_pb (Results 1 – 2 of 2) sorted by relevance
257 std::vector<int64_t> ShapeProtoToVec(const TensorShapeProto& shape_pb) { in ShapeProtoToVec() argument259 if (shape_pb.dim_size() == 0 && !shape_pb.unknown_rank()) { in ShapeProtoToVec()263 for (const auto& d : shape_pb.dim()) { in ShapeProtoToVec()271 TensorShapeProto shape_pb; in VecToShapeProto() local273 shape_pb.set_unknown_rank(true); in VecToShapeProto()274 return shape_pb; in VecToShapeProto()277 shape_pb.add_dim()->set_size(s); in VecToShapeProto()279 return shape_pb; in VecToShapeProto()
38 std::vector<int64_t> ShapeProtoToVec(const TensorShapeProto& shape_pb);