Searched refs:init_table_node (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 377 NodeDef init_table_node; in SparsifyGatherInternal() local
378 init_table_node.set_op("InitializeTable"); in SparsifyGatherInternal()
379 init_table_node.set_name( in SparsifyGatherInternal()
381 SetNodeAttr("Tkey", key_dtype, &init_table_node); in SparsifyGatherInternal()
382 SetNodeAttr("Tval", data_type, &init_table_node); in SparsifyGatherInternal()
383 init_table_node_names.push_back(init_table_node.name()); in SparsifyGatherInternal()
418 AddNodeInput(hashtable_node.name(), &init_table_node); in SparsifyGatherInternal()
420 AddNodeInput(indices_node.name(), &init_table_node); in SparsifyGatherInternal()
422 AddNodeInput(values_node.name(), &init_table_node); in SparsifyGatherInternal()
442 new_nodes->push_back(init_table_node); in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 427 NodeDef* init_table_node = item.graph.add_node(); in TEST_F() local
428 init_table_node->set_op("InitializeTable"); in TEST_F()
429 init_table_node->set_name("initialize_table"); in TEST_F()
430 SetNodeAttr("Tkey", key_dtype, init_table_node); in TEST_F()
431 SetNodeAttr("Tval", data_dtype, init_table_node); in TEST_F()
432 *init_table_node->add_input() = "hash_table"; in TEST_F()
433 *init_table_node->add_input() = "table_keys"; in TEST_F()
434 *init_table_node->add_input() = "table_values"; in TEST_F()
435 item.init_ops.push_back(init_table_node->name()); in TEST_F()
|