Searched refs:indices_node (Results 1 – 2 of 2) sorted by relevance
358 NodeDef indices_node; in SparsifyGatherInternal() local361 &indices_node); in SparsifyGatherInternal()362 SetNodeAttr("dtype", key_dtype, &indices_node); in SparsifyGatherInternal()420 AddNodeInput(indices_node.name(), &init_table_node); in SparsifyGatherInternal()421 refs[indices_node.name()]++; in SparsifyGatherInternal()439 new_nodes->push_back(indices_node); in SparsifyGatherInternal()
63 Node* indices_node; in AsGraphDefInternal() local64 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.indices(), &indices_node)); in AsGraphDefInternal()76 b->AddDataset(this, {indices_node, value_node, dense_shape_node}, in AsGraphDefInternal()