Searched refs:NodeTypeKey (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision.cc | 508 struct NodeTypeKey : public std::pair<absl::string_view, TypeAttrId> { struct in tensorflow::grappler::__anon4a4f70420111::GraphTypeTopologyView
515 friend H AbslHashValue(H h, const NodeTypeKey& nt) { in AbslHashValue()
530 absl::flat_hash_map<NodeTypeKey, int> node_type_name_to_index_;
567 node_type_name_to_index_.emplace(NodeTypeKey(node.name(), type_attr), in InitializeFromGraph()
604 NodeTypeKey(input_node.name(), input_type_attr)); in InitializeFromGraph()
664 NodeTypeKey(edge.src.node->name(), edge.src.type_attr)); in AddEphemeralEdges()
666 NodeTypeKey(edge.dst.node->name(), edge.dst.type_attr)); in AddEphemeralEdges()
684 NodeTypeKey key(node_name, type_attr); in HasNode()
692 NodeTypeKey key(node_name, type_attr); in GetNode()
708 NodeTypeKey key(node_name, type_attr); in GetNodeIndex()
|