1 //*****************************************************************************
2 // Copyright 2017-2020 Intel Corporation
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
8 // http://www.apache.org/licenses/LICENSE-2.0
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //*****************************************************************************
17 #include "ngraph/op/not_equal.hpp"
19 #include "ngraph/runtime/host_tensor.hpp"
20 #include "ngraph/runtime/reference/not_equal.hpp"
22 NGRAPH_SUPPRESS_DEPRECATED_START
25 using namespace ngraph;
27 // ----------------------------------- v0 --------------------------------------
29 constexpr NodeTypeInfo op::v0::NotEqual::type_info;
31 op::v0::NotEqual::NotEqual(const Output<Node>& arg0,
32 const Output<Node>& arg1,
33 const AutoBroadcastSpec& auto_broadcast)
34 : BinaryElementwiseComparison(arg0, arg1, auto_broadcast)
36 constructor_validate_and_infer_types();
39 shared_ptr<Node> op::v0::NotEqual::clone_with_new_inputs(const OutputVector& new_args) const
41 check_new_args_count(this, new_args);
42 return make_shared<op::v0::NotEqual>(new_args.at(0), new_args.at(1), this->get_autob());
47 template <element::Type_t ET>
48 bool evaluate(const HostTensorPtr& arg0,
49 const HostTensorPtr& arg1,
50 const HostTensorPtr& out,
51 const op::AutoBroadcastSpec& broadcast_spec)
53 runtime::reference::not_equal(arg0->get_data_ptr<ET>(),
54 arg1->get_data_ptr<ET>(),
55 out->get_data_ptr<element::Type_t::boolean>(),
62 bool evaluate_not_equal(const HostTensorPtr& arg0,
63 const HostTensorPtr& arg1,
64 const HostTensorPtr& out,
65 const op::AutoBroadcastSpec& broadcast_spec)
68 out->set_broadcast(broadcast_spec, arg0, arg1, element::boolean);
69 switch (arg0->get_element_type())
71 TYPE_CASE(boolean)(arg0, arg1, out, broadcast_spec);
73 TYPE_CASE(i32)(arg0, arg1, out, broadcast_spec);
75 TYPE_CASE(i64)(arg0, arg1, out, broadcast_spec);
77 TYPE_CASE(u32)(arg0, arg1, out, broadcast_spec);
79 TYPE_CASE(u64)(arg0, arg1, out, broadcast_spec);
81 TYPE_CASE(f16)(arg0, arg1, out, broadcast_spec);
83 TYPE_CASE(f32)(arg0, arg1, out, broadcast_spec);
85 default: rc = false; break;
91 bool op::v0::NotEqual::evaluate(const HostTensorVector& outputs,
92 const HostTensorVector& inputs) const
94 OV_ITT_SCOPED_TASK(itt::domains::nGraphOp, "op::v0::NotEqual::evaluate");
95 return evaluate_not_equal(inputs[0], inputs[1], outputs[0], get_autob());
98 // ----------------------------------- v1 --------------------------------------
100 NGRAPH_RTTI_DEFINITION(op::v1::NotEqual, "NotEqual", 1);
102 op::v1::NotEqual::NotEqual(const Output<Node>& arg0,
103 const Output<Node>& arg1,
104 const AutoBroadcastSpec& auto_broadcast)
105 : BinaryElementwiseComparison(arg0, arg1, auto_broadcast)
107 constructor_validate_and_infer_types();
110 shared_ptr<Node> op::v1::NotEqual::clone_with_new_inputs(const OutputVector& new_args) const
112 check_new_args_count(this, new_args);
113 return make_shared<op::v1::NotEqual>(new_args.at(0), new_args.at(1), this->get_autob());
116 bool op::v1::NotEqual::evaluate(const HostTensorVector& outputs,
117 const HostTensorVector& inputs) const
119 OV_ITT_SCOPED_TASK(itt::domains::nGraphOp, "op::v1::NotEqual::evaluate");
120 return evaluate_not_equal(inputs[0], inputs[1], outputs[0], get_autob());