| 798 | } |
| 799 | |
| 800 | void ConvertTanhOperator(const TanhOperator& src_op, |
| 801 | GraphDef* tensorflow_graph) { |
| 802 | tensorflow::NodeDef* tanh_op = tensorflow_graph->add_node(); |
| 803 | tanh_op->set_op("Tanh"); |
| 804 | tanh_op->set_name(src_op.outputs[0]); |
| 805 | *tanh_op->add_input() = src_op.inputs[0]; |
| 806 | (*tanh_op->mutable_attr())["T"].set_type(DT_FLOAT); |
| 807 | } |
| 808 | |
| 809 | void ConvertSoftmaxOperator(const Model& model, const SoftmaxOperator& src_op, |
| 810 | GraphDef* tensorflow_graph) { |
no test coverage detected