Searched refs:NodeTypeKey (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision.cc | 442 struct NodeTypeKey : public std::pair<absl::string_view, TypeAttrId> { struct in tensorflow::grappler::__anon69ea7d4c0111::GraphTypeTopologyView
449 friend H AbslHashValue(H h, const NodeTypeKey& nt) { in AbslHashValue()
464 absl::flat_hash_map<NodeTypeKey, int> node_type_name_to_index_;
501 node_type_name_to_index_.emplace(NodeTypeKey(node.name(), type_attr), in InitializeFromGraph()
538 NodeTypeKey(input_node.name(), input_type_attr)); in InitializeFromGraph()
598 NodeTypeKey(edge.src.node->name(), edge.src.type_attr)); in AddEphemeralEdges()
600 NodeTypeKey(edge.dst.node->name(), edge.dst.type_attr)); in AddEphemeralEdges()
618 NodeTypeKey key(node_name, type_attr); in HasNode()
626 NodeTypeKey key(node_name, type_attr); in GetNode()
642 NodeTypeKey key(node_name, type_attr); in GetNodeIndex()
|