Commit 198102a4 authored by Khalique's avatar Khalique
Browse files

change to generic op

parent 34c61ed3
......@@ -108,6 +108,7 @@ struct tf_parser
{
add_generic_op("Identity", op::identity{});
add_generic_op("Relu", op::relu{});
add_generic_op("Relu6", op::clip{6.0, 0.0});
add_binary_op("Add", op::add{});
add_binary_op("Mul", op::mul{});
......@@ -123,7 +124,6 @@ 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);
......@@ -483,15 +483,6 @@ 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)
{
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment