Searched refs:gather_node (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | sparsify_gather.cc | 294 const NodeDef& gather_node = match.node; in SparsifyGatherInternal() local 298 if (gather_node.op() == "GatherV2") { in SparsifyGatherInternal() 387 lookup_node.set_name(StrCat(gather_node.name(), "/LookupTableFind")); in SparsifyGatherInternal() 395 CreateConstNode(zero_tensor, StrCat(gather_node.name(), "/Const"), in SparsifyGatherInternal() 405 StrCat(gather_node.name(), "/ExpandDims/Const")); in SparsifyGatherInternal() 413 expand_dims_node.set_name(gather_node.name()); in SparsifyGatherInternal() 426 AddNodeInput(gather_node.input(1), &lookup_node); in SparsifyGatherInternal() 427 refs[gather_node.input(1)]++; in SparsifyGatherInternal()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | arithmetic_optimizer.cc | 3542 NodeDef* gather_node = nullptr; in TrySimplify() local 3543 TF_RETURN_IF_ERROR(GetInputNode(reduction_node->input(0), &gather_node)); in TrySimplify() 3544 if (!IsGather(*gather_node) || IsInPreserveSet(*gather_node) || in TrySimplify() 3545 gather_node->device() != reduction_node->device()) in TrySimplify() 3547 if (gather_node->op() == "GatherV2" && !IsAxis0(*gather_node, 2)) in TrySimplify() 3553 TF_RETURN_IF_ERROR(GetInputNode(gather_node->input(1), &unique_node)); in TrySimplify() 3555 unique_node->device() != gather_node->device()) in TrySimplify() 3570 reduction_node->set_input(0, gather_node->input(0)); in TrySimplify() 3573 gather_node->input(0)); in TrySimplify()
|