Searched refs:reduction_node (Results 1 – 1 of 1) sorted by relevance
2787 Status TrySimplify(NodeDef* reduction_node, in TrySimplify() argument2789 if (IsInPreserveSet(*reduction_node)) { in TrySimplify()2793 TF_RETURN_IF_ERROR(GetInputNode(reduction_node->input(0), &inner_function)); in TrySimplify()2804 (is_non_decreasing || !IsAnyMaxPool(*reduction_node))) { in TrySimplify()2808 reduction_node->set_input(0, inner_input->name()); in TrySimplify()2809 ctx().node_map->UpdateInput(reduction_node->name(), in TrySimplify()2811 inner_function->set_input(0, reduction_node->name()); in TrySimplify()2812 UpdateConsumers(reduction_node, inner_function->name()); in TrySimplify()2814 reduction_node->name()); in TrySimplify()2818 const string opposite = FlipMinMax(*reduction_node); in TrySimplify()[all …]