| 771 | } |
| 772 | |
| 773 | void ConvertRelu6Operator(const Relu6Operator& src_op, |
| 774 | GraphDef* tensorflow_graph) { |
| 775 | tensorflow::NodeDef* relu_op = tensorflow_graph->add_node(); |
| 776 | relu_op->set_op("Relu6"); |
| 777 | relu_op->set_name(src_op.outputs[0]); |
| 778 | *relu_op->add_input() = src_op.inputs[0]; |
| 779 | (*relu_op->mutable_attr())["T"].set_type(DT_FLOAT); |
| 780 | } |
| 781 | |
| 782 | void ConvertLogOperator(const LogOperator& src_op, GraphDef* tensorflow_graph) { |
| 783 | tensorflow::NodeDef* op = tensorflow_graph->add_node(); |
no test coverage detected