Searched refs:unique_node (Results 1 – 2 of 2) sorted by relevance
3068 std::unordered_set<Node*> unique_node; in MergeConv2DWithBiasAdd() local3071 auto result = unique_node.insert(e->src()); in MergeConv2DWithBiasAdd()3077 unique_node.clear(); in MergeConv2DWithBiasAdd()3081 auto result = unique_node.insert(e->src()); in MergeConv2DWithBiasAdd()3087 unique_node.clear(); in MergeConv2DWithBiasAdd()3093 auto result = unique_node.insert(e->dst()); in MergeConv2DWithBiasAdd()3099 unique_node.clear(); in MergeConv2DWithBiasAdd()3104 auto result = unique_node.insert(e->dst()); in MergeConv2DWithBiasAdd()3370 std::unordered_set<Node*> unique_node; in MergeConv2DBackpropFilterWithBiasAddGrad() local3373 auto result = unique_node.insert(e->src()); in MergeConv2DBackpropFilterWithBiasAddGrad()[all …]
3552 NodeDef* unique_node = nullptr; in TrySimplify() local3553 TF_RETURN_IF_ERROR(GetInputNode(gather_node->input(1), &unique_node)); in TrySimplify()3554 if (!IsUnique(*unique_node) || IsInPreserveSet(*unique_node) || in TrySimplify()3555 unique_node->device() != gather_node->device()) in TrySimplify()3557 if (unique_node->op() == "UniqueV2" && !IsAxis0(*unique_node, 1)) in TrySimplify()3561 TF_RETURN_IF_ERROR(GetNodeAttr(*unique_node, "T", &unique_element_type)); in TrySimplify()3566 if (idx_tensor != TensorId(unique_node->name(), 1)) return Status::OK(); in TrySimplify()3577 reduction_node->set_input(1, unique_node->input(0)); in TrySimplify()3580 unique_node->input(0)); in TrySimplify()