Searched refs:values_node (Results 1 – 2 of 2) sorted by relevance
364 NodeDef values_node; in SparsifyGatherInternal() local366 &values_node); in SparsifyGatherInternal()367 SetNodeAttr("dtype", data_type, &values_node); in SparsifyGatherInternal()422 AddNodeInput(values_node.name(), &init_table_node); in SparsifyGatherInternal()423 refs[values_node.name()]++; in SparsifyGatherInternal()440 new_nodes->push_back(values_node); in SparsifyGatherInternal()
417 NodeDef* values_node = item.graph.add_node(); in TEST_F() local418 values_node->set_op("Const"); in TEST_F()419 values_node->set_name("table_values"); in TEST_F()420 SetNodeAttr("dtype", data_dtype, values_node); in TEST_F()424 SetNodeAttr("value", values, values_node); in TEST_F()