Searched refs:hashtable_node (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 369 NodeDef hashtable_node; in SparsifyGatherInternal() local
370 hashtable_node.set_op("HashTable"); in SparsifyGatherInternal()
371 hashtable_node.set_name(StrCat(weights_node.name(), "/HashTable")); in SparsifyGatherInternal()
372 SetNodeAttr("key_dtype", key_dtype, &hashtable_node); in SparsifyGatherInternal()
373 SetNodeAttr("value_dtype", data_type, &hashtable_node); in SparsifyGatherInternal()
417 AddNodeInput(hashtable_node.name(), &init_table_node); in SparsifyGatherInternal()
418 refs[hashtable_node.name()]++; in SparsifyGatherInternal()
424 AddNodeInput(hashtable_node.name(), &lookup_node); in SparsifyGatherInternal()
425 refs[hashtable_node.name()]++; in SparsifyGatherInternal()
440 new_nodes->push_back(hashtable_node); in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 415 NodeDef* hashtable_node = item.graph.add_node(); in TEST_F() local
416 hashtable_node->set_op("HashTable"); in TEST_F()
417 hashtable_node->set_name("hash_table"); in TEST_F()
418 SetNodeAttr("key_dtype", key_dtype, hashtable_node); in TEST_F()
419 SetNodeAttr("value_dtype", data_dtype, hashtable_node); in TEST_F()
|