1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
//
// This confidential and proprietary software may be used only as
// authorised by a licensing agreement from ARM Limited
// (C) COPYRIGHT 2020-2022 ARM Limited
// ALL RIGHTS RESERVED
// The entire notice above must be reproduced on all authorised
// copies and copies may only be made to the extent permitted
// by a licensing agreement from ARM Limited.
=== Comparison Operators
==== EQUAL
Elementwise comparison operation
*Arguments:*
|===
|Argument|Type|Name|Shape|Description
|Input|in_t*|input1|shape1|Input tensor
|Input|in_t*|input2|shape2|Input tensor with the same rank as input1
|Output|out_t*|output|shape|Output tensor with broadcast shape if necessary
|===
*Operation Function:*
[source,c++]
----
for_each(index in shape) {
index1 = apply_broadcast(shape, shape1, index);
index2 = apply_broadcast(shape, shape2, index);
in_t value1 = tensor_read<in_t>(input1, shape1, index1);
in_t value2 = tensor_read<in_t>(input2, shape2, index2);
out_t result;
if (isNaN(value1) || isNaN(value2))
result = False;
else
result = (value1 == value2) ? True : False;
tensor_write<out_t>(output, shape, index, result);
}
----
*Supported Data Types:*
|===
|Profile|Mode|in_t|out_t
|Any|signed 32|int32_t|bool_t
|MI, MT|fp16|fp16_t|bool_t
|MI, MT|bf16|bf16_t|bool_t
|MI, MT|fp32|fp32_t|bool_t
|===
==== GREATER
Elementwise greater than comparison operation
*Arguments:*
|===
|Argument|Type|Name|Shape|Description
|Input|in_t*|input1|shape1|Input tensor
|Input|in_t*|input2|shape2|Input tensor with the same rank as input1
|Output|out_t*|output|shape|Output tensor with broadcast shape if necessary
|===
*Operation Function:*
[source,c++]
----
for_each(index in shape) {
index1 = apply_broadcast(shape, shape1, index);
index2 = apply_broadcast(shape, shape2, index);
in_t value1 = tensor_read<in_t>(input1, shape1, index1);
in_t value2 = tensor_read<in_t>(input2, shape2, index2);
out_t result;
if (isNaN(value1) || isNaN(value2))
result = False;
else
result = (value1 > value2) ? True : False;
tensor_write<out_t>(output, shape, index, result);
}
----
*Supported Data Types:*
|===
|Profile|Mode|in_t|out_t
|Any|signed 32|int32_t|bool_t
|MI, MT|fp16|fp16_t|bool_t
|MI, MT|bf16|bf16_t|bool_t
|MI, MT|fp32|fp32_t|bool_t
|===
==== GREATER_EQUAL
Elementwise comparison operation
*Arguments:*
|===
|Argument|Type|Name|Shape|Description
|Input|in_t*|input1|shape1|Input tensor
|Input|in_t*|input2|shape2|Input tensor with the same rank as input1
|Output|out_t*|output|shape|Output tensor with broadcast shape if necessary
|===
*Operation Function:*
[source,c++]
----
for_each(index in shape) {
index1 = apply_broadcast(shape, shape1, index);
index2 = apply_broadcast(shape, shape2, index);
in_t value1 = tensor_read<in_t>(input1, shape1, index1);
in_t value2 = tensor_read<in_t>(input2, shape2, index2);
out_t result;
if (isNaN(value1) || isNaN(value2))
result = False;
else
result = (value1 >= value2) ? True : False;
tensor_write<out_t>(output, shape, index, result);
}
----
*Supported Data Types:*
|===
|Profile|Mode|in_t|out_t
|Any|signed 32|int32_t|bool_t
|MI, MT|fp16|fp16_t|bool_t
|MI, MT|bf16|bf16_t|bool_t
|MI, MT|fp32|fp32_t|bool_t
|===
|
