| 789 | } |
| 790 | |
| 791 | void ConvertLogisticOperator(const LogisticOperator& src_op, |
| 792 | GraphDef* tensorflow_graph) { |
| 793 | tensorflow::NodeDef* relu_op = tensorflow_graph->add_node(); |
| 794 | relu_op->set_op("Sigmoid"); |
| 795 | relu_op->set_name(src_op.outputs[0]); |
| 796 | *relu_op->add_input() = src_op.inputs[0]; |
| 797 | (*relu_op->mutable_attr())["T"].set_type(DT_FLOAT); |
| 798 | } |
| 799 | |
| 800 | void ConvertTanhOperator(const TanhOperator& src_op, |
| 801 | GraphDef* tensorflow_graph) { |
no test coverage detected