def equal(left_node, right_node, name=None): # type: (NodeInput, NodeInput, str) -> Node
"""Return node which checks if input nodes are equal element-wise.
:param left_node: The first input node for equal operation.
:param right_node: The second input node for equal operation.
:param name: The optional name for output new node.
:return: The node performing element-wise equality check.
"""
return Equal(left_node, right_node)
def binary_op(op_str, a, b):
if op_str == '+':
return a + b
elif op_str == 'Add':
return Add(a, b)
elif op_str == '-':
return a - b
elif op_str == 'Sub':
return Subtract(a, b)
elif op_str == '*':
return a * b
elif op_str == 'Mul':
return Multiply(a, b)
elif op_str == '/':
return a / b
elif op_str == 'Div':
return Divide(a, b)
elif op_str == 'Dot':
return Dot(a, b)
elif op_str == 'Equal':
return Equal(a, b)
elif op_str == 'Greater':
return Greater(a, b)
elif op_str == 'GreaterEq':
return GreaterEq(a, b)
elif op_str == 'Less':
return Less(a, b)
elif op_str == 'LessEq':
return LessEq(a, b)
elif op_str == 'Maximum':
return Maximum(a, b)
elif op_str == 'Minimum':
return Minimum(a, b)
elif op_str == 'NotEqual':
return NotEqual(a, b)
elif op_str == 'Power':
return Power(a, b)
def equal(left_node, right_node, name=None): # type: (NodeInput, NodeInput, str) -> Node
"""Return node which checks if input nodes are equal elementwise."""
return Equal(left_node, right_node)