| 1801 | } |
| 1802 | |
| 1803 | Status SavedModelOptimizer::ConvertToHashTableOp( |
| 1804 | Node* node, std::vector<SrcInfo>& input_info) { |
| 1805 | std::unordered_map<std::string, std::string> attr_info_map; |
| 1806 | |
| 1807 | // KvVarHandleOp -> MutableHashTableV2 |
| 1808 | attr_info_map["container"] = "container"; |
| 1809 | attr_info_map["shared_name"] = "shared_name"; |
| 1810 | attr_info_map["dtype"] = "value_dtype"; |
| 1811 | attr_info_map["Tkeys"] = "key_dtype"; |
| 1812 | attr_info_map["shape"] = "value_shape"; |
| 1813 | |
| 1814 | return ReplaceNode( |
| 1815 | "MutableHashTableOfTensorsV2", node, &graph_, input_info, |
| 1816 | attr_info_map); |
| 1817 | } |
| 1818 | |
| 1819 | Status SavedModelOptimizer::ConvertToHashLookupOp( |
| 1820 | Node* node, std::vector<SrcInfo>& input_info) { |
nothing calls this directly
no test coverage detected