add relu6 op and tests

src/tf/tf.cpp

@@ -123,6 +123,7 @@ struct tf_parser
         add_mem_op("Mean", &tf_parser::parse_mean);
         add_mem_op("Pack", &tf_parser::parse_pack);
         add_mem_op("Pad", &tf_parser::parse_pad);
+        add_mem_op("Relu6", &tf_parser::parse_relu6);
         add_mem_op("Reshape", &tf_parser::parse_reshape);
         add_mem_op("Softmax", &tf_parser::parse_softmax);
         add_mem_op("Squeeze", &tf_parser::parse_squeeze);
@@ -482,6 +483,15 @@ struct tf_parser
         return prog.add_instruction(op, args[0]);
     }
 
+    instruction_ref
+    parse_relu6(const std::string&, const attribute_map&, std::vector<instruction_ref> args)
+    {
+        op::clip op;
+        op.max_val = 6.0;
+        op.min_val = 0.0;
+        return prog.add_instruction(op, std::move(args));
+    }
+
     instruction_ref
     parse_reshape(const std::string&, const attribute_map&, std::vector<instruction_ref> args)
     {

test/tf/relu6_test.pb

+
+:
+
0Placeholder*
+dtype0
*
+shape:

+
+relu6Relu6
0*
+
T0
"
\ No newline at end of file

test/tf/tf_test.cpp

@@ -229,6 +229,16 @@ TEST_CASE(relu_test)
     EXPECT(p == prog);
 }
 
+TEST_CASE(relu6_test)
+{
+    migraphx::program p;
+    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
+    p.add_instruction(migraphx::op::clip{6.0, 0.0}, l0);
+    auto prog = migraphx::parse_tf("relu6_test.pb", false);
+
+    EXPECT(p == prog);
+}
+
 TEST_CASE(reshape_test)
 {
     migraphx::program p;