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()
|