Searched refs:gather_node (Results 1 – 1 of 1) sorted by relevance
295 const NodeDef& gather_node = match.node; in SparsifyGatherInternal() local299 if (gather_node.op() == "GatherV2") { in SparsifyGatherInternal()388 lookup_node.set_name(StrCat(gather_node.name(), "/LookupTableFind")); in SparsifyGatherInternal()396 CreateConstNode(zero_tensor, StrCat(gather_node.name(), "/Const"), in SparsifyGatherInternal()406 StrCat(gather_node.name(), "/ExpandDims/Const")); in SparsifyGatherInternal()414 expand_dims_node.set_name(gather_node.name()); in SparsifyGatherInternal()427 AddNodeInput(gather_node.input(1), &lookup_node); in SparsifyGatherInternal()428 refs[gather_node.input(1)]++; in SparsifyGatherInternal()