Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc376 NodeDef init_table_node; in SparsifyGatherInternal() local
377 init_table_node.set_op("InitializeTable"); in SparsifyGatherInternal()
378 init_table_node.set_name( in SparsifyGatherInternal()
380 SetNodeAttr("Tkey", key_dtype, &init_table_node); in SparsifyGatherInternal()
381 SetNodeAttr("Tval", data_type, &init_table_node); in SparsifyGatherInternal()
382 init_table_node_names.push_back(init_table_node.name()); in SparsifyGatherInternal()
417 AddNodeInput(hashtable_node.name(), &init_table_node); in SparsifyGatherInternal()
419 AddNodeInput(indices_node.name(), &init_table_node); in SparsifyGatherInternal()
421 AddNodeInput(values_node.name(), &init_table_node); in SparsifyGatherInternal()
441 new_nodes->push_back(init_table_node); in SparsifyGatherInternal()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc441 NodeDef* init_table_node = item.graph.add_node(); in TEST_F() local
442 init_table_node->set_op("InitializeTable"); in TEST_F()
443 init_table_node->set_name("initialize_table"); in TEST_F()
444 SetNodeAttr("Tkey", key_dtype, init_table_node); in TEST_F()
445 SetNodeAttr("Tval", data_dtype, init_table_node); in TEST_F()
446 *init_table_node->add_input() = "hash_table"; in TEST_F()
447 *init_table_node->add_input() = "table_keys"; in TEST_F()
448 *init_table_node->add_input() = "table_values"; in TEST_F()
449 item.init_ops.push_back(init_table_node->name()); in TEST_F()