Searched refs:default_value_node (Results 1 – 2 of 2) sorted by relevance
394 NodeDef default_value_node; in SparsifyGatherInternal() local396 &default_value_node); in SparsifyGatherInternal()397 SetNodeAttr("dtype", data_type, &default_value_node); in SparsifyGatherInternal()428 AddNodeInput(default_value_node.name(), &lookup_node); in SparsifyGatherInternal()429 refs[default_value_node.name()]++; in SparsifyGatherInternal()443 new_nodes->push_back(default_value_node); in SparsifyGatherInternal()
461 NodeDef* default_value_node = item.graph.add_node(); in TEST_F() local462 default_value_node->set_op("Const"); in TEST_F()463 default_value_node->set_name("default_table_value"); in TEST_F()464 SetNodeAttr("dtype", data_dtype, default_value_node); in TEST_F()467 SetNodeAttr("value", dflt, default_value_node); in TEST_F()