Skip to content

Commit 85c8f5d

Browse files
committed
[circle-mlir] Introduce test model for Sign operation
This introduces the test models for Sign operation. It will be activated after the operation is implemented. ONE-DCO-1.0-Signed-off-by: Seungho Henry Park <shs.park@samsung.com>
1 parent 9adb851 commit 85c8f5d

File tree

4 files changed

+22
-0
lines changed

4 files changed

+22
-0
lines changed

circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,7 @@ AddModel(Resize_F32_R4_nearest)
170170
AddModel(Shape_F32_R4)
171171
AddModel(Sigmoid_F32_R2)
172172
AddModel(Sigmoid_F32_R4)
173+
# AddModel(Sign_F32_R4) # TODO activate after supporting this operation
173174
AddModel(Sin_F32_R4)
174175
AddModel(Slice_F32_R2_4) # Rank2 of Slice_F32_R4_4
175176
AddModel(Slice_F32_R3_4) # Rank3 of Slice_F32_R4_4

circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,7 @@ AddModel(Rsqrt_F32_R4)
179179
AddModel(Shape_F32_R4)
180180
AddModel(Sigmoid_F32_R2)
181181
AddModel(Sigmoid_F32_R4)
182+
# AddModel(Sign_F32_R4) # TODO activate after supporting this operation
182183
AddModel(Sin_F32_R4)
183184
AddModel(Slice_F32_R2_4) # Rank2 of Slice_F32_R4_4
184185
AddModel(Slice_F32_R3_4) # Rank3 of Slice_F32_R4_4

circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,7 @@ AddModel(Rsqrt_F32_R4)
179179
AddModel(Shape_F32_R4)
180180
AddModel(Sigmoid_F32_R2)
181181
AddModel(Sigmoid_F32_R4)
182+
# AddModel(Sign_F32_R4) # TODO activate after supporting this operation
182183
AddModel(Sin_F32_R4)
183184
AddModel(Slice_F32_R2_4) # Rank2 of Slice_F32_R4_4
184185
AddModel(Slice_F32_R3_4) # Rank3 of Slice_F32_R4_4
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
import torch
2+
3+
4+
# Generate Sign operator with Float32, Rank-4
5+
class net_Sign(torch.nn.Module):
6+
def __init__(self):
7+
super().__init__()
8+
9+
def forward(self, input):
10+
return torch.sign(input)
11+
12+
def onnx_opset_version(self):
13+
# TODO set to appropriate value
14+
return 14
15+
16+
17+
_model_ = net_Sign()
18+
19+
_inputs_ = torch.randn(1, 2, 3, 3)

0 commit comments

Comments
 (0)