Skip to content

GitLab

Commit 9b7c44ab authored by Paul's avatar Paul
Browse files

Formatting

parent 552ad02c
Hide whitespace changes
Inline Side-by-side
Showing with 15 additions and 13 deletions
src/tf/tf.cpp
View file @ 9b7c44ab
...@@ -204,19 +204,21 @@ struct tf_parser ...@@ -204,19 +204,21 @@ struct tf_parser
template <class T> template <class T>
void add_binary_op(std::string name, T x) void add_binary_op(std::string name, T x)
{ {
add_op(name, [this, x](const attribute_map&, std::vector<instruction_ref> args) { add_op(name,
if(args.size() != 2) [this, x](const attribute_map&, std::vector<instruction_ref> args) {
MIGRAPHX_THROW("binary operators should have 2 operands"); if(args.size() != 2)
// TODO MIGRAPHX_THROW("binary operators should have 2 operands");
// if(contains(attributes, "data_format")) // TODO
// { // if(contains(attributes, "data_format"))
// if(is_nhwc) // {
// { // if(is_nhwc)
// l0 = prog.add_instruction(op::transpose{{0, 3, 1, 2}}, args[1]); // {
// } // l0 = prog.add_instruction(op::transpose{{0, 3, 1, 2}}, args[1]);
// } // }
return add_broadcastable_binary_op(args[0], args[1], x); // }
}, false); return add_broadcastable_binary_op(args[0], args[1], x);
},
false);
} }
template <class T> template <class T>
......
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