Home
last modified time | relevance | path

Searched refs:reciprocal_node (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2789 NodeDef* reciprocal_node = optimized_graph->add_node(); in ReduceDivToReciprocalMul() local
2790 reciprocal_node->set_name(OptimizedNodeName(*node, "_recip")); in ReduceDivToReciprocalMul()
2791 reciprocal_node->set_op("Reciprocal"); in ReduceDivToReciprocalMul()
2792 reciprocal_node->set_device(node->device()); in ReduceDivToReciprocalMul()
2795 reciprocal_node->add_input(const_input); in ReduceDivToReciprocalMul()
2796 (*reciprocal_node->mutable_attr())["T"].set_type(type); in ReduceDivToReciprocalMul()
2797 node->set_input(1, reciprocal_node->name()); in ReduceDivToReciprocalMul()
2798 node_map_->AddNode(reciprocal_node->name(), reciprocal_node); in ReduceDivToReciprocalMul()
2799 node_map_->UpdateOutput(node->name(), const_input, reciprocal_node->name()); in ReduceDivToReciprocalMul()