Searched refs:reduction_node (Results 1 – 1 of 1) sorted by relevance
3475 Status TrySimplify(NodeDef* reduction_node, in TrySimplify() argument3477 if (IsInPreserveSet(*reduction_node)) { in TrySimplify()3482 TF_RETURN_IF_ERROR(GetInputNode(reduction_node->input(0), &inner_function)); in TrySimplify()3511 (is_non_decreasing || !IsAnyMaxPool(*reduction_node)) && in TrySimplify()3516 reduction_node->set_input(0, inner_input->name()); in TrySimplify()3517 ctx().node_map->UpdateInput(reduction_node->name(), in TrySimplify()3519 inner_function->set_input(0, reduction_node->name()); in TrySimplify()3521 UpdateConsumers(reduction_node, inner_function->name())); in TrySimplify()3523 reduction_node->name()); in TrySimplify()3527 const string opposite = FlipMinMax(*reduction_node); in TrySimplify()[all …]