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.cc3068 NodeDef* reciprocal_node = optimized_graph->add_node(); in ReduceDivToReciprocalMul() local
3069 reciprocal_node->set_name(OptimizedNodeName(*node, "_recip")); in ReduceDivToReciprocalMul()
3070 reciprocal_node->set_op("Reciprocal"); in ReduceDivToReciprocalMul()
3071 reciprocal_node->set_device(node->device()); in ReduceDivToReciprocalMul()
3072 reciprocal_node->add_input(const_input); in ReduceDivToReciprocalMul()
3073 (*reciprocal_node->mutable_attr())["T"].set_type(type); in ReduceDivToReciprocalMul()
3079 node->set_input(0, reciprocal_node->name()); in ReduceDivToReciprocalMul()
3082 node->set_input(1, reciprocal_node->name()); in ReduceDivToReciprocalMul()
3084 node_map_->AddNode(reciprocal_node->name(), reciprocal_node); in ReduceDivToReciprocalMul()
3085 node_map_->UpdateOutput(node->name(), const_input, reciprocal_node->name()); in ReduceDivToReciprocalMul()