| 18 | class BinaryNode : public NaryNode { |
| 19 | public: |
| 20 | BinaryNode(const af::dtype type, const char *op_str, common::Node_ptr lhs, |
| 21 | common::Node_ptr rhs, af_op_t op) |
| 22 | : NaryNode(type, op_str, 2, {{lhs, rhs}}, op, |
| 23 | std::max(lhs->getHeight(), rhs->getHeight()) + 1) {} |
| 24 | }; |
| 25 | |
| 26 | template<typename To, typename Ti, af_op_t op> |