Home
last modified time | relevance | path

Searched refs:NodeClass (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph.h299 enum NodeClass { enum
334 NodeClass node_class);
336 static NodeClass GetNodeClassForOp(const std::string& ts);
340 NodeClass class_;
764 const Node* cost_node, Node::NodeClass node_class);
Dgraph.cc45 Node::NodeClass Node::GetNodeClassForOp(const std::string& ts) { in GetNodeClassForOp()
46 static const absl::flat_hash_map<std::string, Node::NodeClass>* in GetNodeClassForOp()
50 new absl::flat_hash_map<std::string, Node::NodeClass>({ in GetNodeClassForOp()
135 Node::NodeClass node_class) { in Initialize()
509 Node::NodeClass node_class = op_reg_data->is_function_op in AddNode()
870 const Node* cost_node, Node::NodeClass node_class) { in AllocateNode()
/external/tensorflow/tensorflow/core/framework/
Dmodel.proto9 enum NodeClass { enum
97 NodeClass node_class = 15;
Dmodel.cc487 node_proto->set_node_class(NodeClass::INTERLEAVE_MANY); in ToProto()
710 node_proto->set_node_class(NodeClass::ASYNC_INTERLEAVE_MANY); in ToProto()
802 node_proto->set_node_class(NodeClass::KNOWN_RATIO); in ToProto()
1169 node_proto->set_node_class(NodeClass::UNKNOWN_RATIO); in ToProto()
1223 node_proto->set_node_class(NodeClass::UNKNOWN); in ToProto()
1250 node_proto->set_node_class(NodeClass::ASYNC_KNOWN_RATIO); in ToProto()
1295 node_proto->set_node_class(NodeClass::ASYNC_UNKNOWN_RATIO); in ToProto()
2087 case NodeClass::INTERLEAVE_MANY: in FromProto()
2090 case NodeClass::ASYNC_INTERLEAVE_MANY: in FromProto()
2094 case NodeClass::KNOWN_RATIO: in FromProto()
[all …]