Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc386 NodeDef lookup_node; in SparsifyGatherInternal() local
387 lookup_node.set_op("LookupTableFind"); in SparsifyGatherInternal()
388 lookup_node.set_name(StrCat(gather_node.name(), "/LookupTableFind")); in SparsifyGatherInternal()
389 SetNodeAttr("Tin", key_dtype, &lookup_node); in SparsifyGatherInternal()
390 SetNodeAttr("Tout", data_type, &lookup_node); in SparsifyGatherInternal()
425 AddNodeInput(hashtable_node.name(), &lookup_node); in SparsifyGatherInternal()
427 AddNodeInput(gather_node.input(1), &lookup_node); in SparsifyGatherInternal()
429 AddNodeInput(default_value_node.name(), &lookup_node); in SparsifyGatherInternal()
432 AddNodeInput(lookup_node.name(), &expand_dims_node); in SparsifyGatherInternal()
433 refs[lookup_node.name()]++; in SparsifyGatherInternal()
[all …]
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc456 NodeDef* lookup_node = item.graph.add_node(); in TEST_F() local
457 lookup_node->set_op("LookupTableFind"); in TEST_F()
458 lookup_node->set_name("table_lookup"); in TEST_F()
459 SetNodeAttr("Tin", key_dtype, lookup_node); in TEST_F()
460 SetNodeAttr("Tout", data_dtype, lookup_node); in TEST_F()
461 *lookup_node->add_input() = "hash_table"; in TEST_F()
462 *lookup_node->add_input() = "query"; in TEST_F()
463 *lookup_node->add_input() = "default_table_value"; in TEST_F()
464 item.fetch.push_back(lookup_node->name()); in TEST_F()
/external/python/cpython2/Lib/compiler/
Dtransformer.py365 exprNode = self.lookup_node(en)(en[1:])
715 node = self.lookup_node(nodelist[-1])(nodelist[-1][1:])
797 def lookup_node(self, node): member in Transformer
1082 return self.lookup_node(n)(n[1:])
1086 items.append(self.lookup_node(n)(n[1:]))
1090 result = self.lookup_node(node)(node[1:])
1097 result = self.lookup_node(node)(node[1:])