Home
last modified time | relevance | path

Searched refs:node_proto (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dmodel.cc485 Status ToProto(ModelProto::Node* node_proto) const { in ToProto()
486 TF_RETURN_IF_ERROR(Node::ToProto(node_proto)); in ToProto()
487 node_proto->set_node_class(NodeClass::INTERLEAVE_MANY); in ToProto()
708 Status ToProto(ModelProto::Node* node_proto) const { in ToProto()
709 TF_RETURN_IF_ERROR(Node::ToProto(node_proto)); in ToProto()
710 node_proto->set_node_class(NodeClass::ASYNC_INTERLEAVE_MANY); in ToProto()
800 Status ToProto(ModelProto::Node* node_proto) const { in ToProto()
801 TF_RETURN_IF_ERROR(Node::ToProto(node_proto)); in ToProto()
802 node_proto->set_node_class(NodeClass::KNOWN_RATIO); in ToProto()
803 node_proto->set_ratio(ratio_); in ToProto()
[all …]
Dmodel.h466 virtual Status ToProto(ModelProto::Node* node_proto) const;
469 static Status FromProto(ModelProto::Node node_proto,
660 static Status FromProtoHelper(ModelProto::Node node_proto,
/external/pytorch/torch/csrc/jit/serialization/
Dexport.cpp319 onnx::NodeProto* node_proto,
373 onnx::NodeProto* node_proto,
379 onnx::NodeProto* node_proto,
925 onnx::NodeProto* node_proto, in EncodeNode() argument
931 node_proto->set_doc_string(node->sourceRange().str()); in EncodeNode()
935 node_proto->add_input(""); in EncodeNode()
937 node_proto->add_input(input->debugName()); in EncodeNode()
941 node_proto->add_output(output->debugName()); in EncodeNode()
953 node_proto->set_domain(domain); in EncodeNode()
960 node_proto->set_op_type(node->kind().toUnqualString()); in EncodeNode()
[all …]
/external/pytorch/torch/utils/tensorboard/
D_proto_graph.py32 def node_proto( function
D_pytorch_graph.py12 from ._proto_graph import node_proto
222 node_proto(