Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc244 const auto iter = type2io_.find(&node); in GetTypeAttrs()
245 CHECK(iter != type2io_.end()); // Crash Ok in GetTypeAttrs()
255 return type2io_.at(&node).at(type_attr).first; in GetInputPorts()
261 return type2io_.at(&node).at(type_attr).second; in GetOutputPorts()
285 auto& type2io_entry = type2io_[&node]; in AddNode()
357 absl::flat_hash_map<const NodeDef*, Type2IOMap> type2io_; member in tensorflow::grappler::__anon69ea7d4c0111::NodeTypeAttrMap