onnx/defs/printer.h

0001 /*
0002  * SPDX-License-Identifier: Apache-2.0
0003  */
0004
0005 #pragma once
0006
0007 #include <iostream>
0008 #include <string>
0009
0010 #include "onnx/defs/parser.h"
0011 #include "onnx/onnx_pb.h"
0012
0013 namespace ONNX_NAMESPACE {
0014
0015 std::ostream& operator<<(std::ostream& os, const TensorShapeProto_Dimension& dim);
0016
0017 std::ostream& operator<<(std::ostream& os, const TensorShapeProto& shape);
0018
0019 std::ostream& operator<<(std::ostream& os, const TypeProto_Tensor& tensortype);
0020
0021 std::ostream& operator<<(std::ostream& os, const TypeProto& type);
0022
0023 std::ostream& operator<<(std::ostream& os, const TensorProto& tensor);
0024
0025 std::ostream& operator<<(std::ostream& os, const ValueInfoProto& value_info);
0026
0027 std::ostream& operator<<(std::ostream& os, const ValueInfoList& vilist);
0028
0029 std::ostream& operator<<(std::ostream& os, const AttributeProto& attr);
0030
0031 std::ostream& operator<<(std::ostream& os, const AttrList& attrlist);
0032
0033 std::ostream& operator<<(std::ostream& os, const NodeProto& node);
0034
0035 std::ostream& operator<<(std::ostream& os, const NodeList& nodelist);
0036
0037 std::ostream& operator<<(std::ostream& os, const GraphProto& graph);
0038
0039 std::ostream& operator<<(std::ostream& os, const FunctionProto& fn);
0040
0041 std::ostream& operator<<(std::ostream& os, const ModelProto& model);
0042
0043 template <typename ProtoType>
0044 std::string ProtoToString(const ProtoType& proto) {
0045   std::stringstream ss;
0046   ss << proto;
0047   return ss.str();
0048 }
0049
0050 } // namespace ONNX_NAMESPACE