Searched refs:reduction_node (Results 1 – 1 of 1) sorted by relevance
2877 Status TrySimplify(NodeDef* reduction_node, in TrySimplify() argument2879 if (IsInPreserveSet(*reduction_node)) { in TrySimplify()2884 TF_RETURN_IF_ERROR(GetInputNode(reduction_node->input(0), &inner_function)); in TrySimplify()2913 (is_non_decreasing || !IsAnyMaxPool(*reduction_node)) && in TrySimplify()2918 reduction_node->set_input(0, inner_input->name()); in TrySimplify()2919 ctx().node_map->UpdateInput(reduction_node->name(), in TrySimplify()2921 inner_function->set_input(0, reduction_node->name()); in TrySimplify()2922 UpdateConsumers(reduction_node, inner_function->name()); in TrySimplify()2924 reduction_node->name()); in TrySimplify()2928 const string opposite = FlipMinMax(*reduction_node); in TrySimplify()[all …]