diff options
Diffstat (limited to 'model-integration/src/test/models')
16 files changed, 0 insertions, 318 deletions
diff --git a/model-integration/src/test/models/onnx/add_double.onnx b/model-integration/src/test/models/onnx/add_double.onnx deleted file mode 100644 index 9264d1eb9f9..00000000000 --- a/model-integration/src/test/models/onnx/add_double.onnx +++ /dev/null @@ -1,16 +0,0 @@ -
add_double.py:f - -input1 -input2output"AddaddZ -input1 - - -Z -input2 - - -b -output - - -B
\ No newline at end of file diff --git a/model-integration/src/test/models/onnx/add_double.py b/model-integration/src/test/models/onnx/add_double.py deleted file mode 100755 index fa9aa48f4b2..00000000000 --- a/model-integration/src/test/models/onnx/add_double.py +++ /dev/null @@ -1,27 +0,0 @@ -# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import onnx -from onnx import helper, TensorProto - -INPUT_1 = helper.make_tensor_value_info('input1', TensorProto.DOUBLE, [1]) -INPUT_2 = helper.make_tensor_value_info('input2', TensorProto.DOUBLE, [1]) -OUTPUT = helper.make_tensor_value_info('output', TensorProto.DOUBLE, [1]) - -nodes = [ - helper.make_node( - 'Add', - ['input1', 'input2'], - ['output'], - ), -] -graph_def = helper.make_graph( - nodes, - 'add', - [ - INPUT_1, - INPUT_2 - ], - [OUTPUT], -) -model_def = helper.make_model(graph_def, producer_name='add_double.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) -onnx.save(model_def, 'add_double.onnx') diff --git a/model-integration/src/test/models/onnx/add_float.onnx b/model-integration/src/test/models/onnx/add_float.onnx deleted file mode 100644 index 0e3ad8f900c..00000000000 --- a/model-integration/src/test/models/onnx/add_float.onnx +++ /dev/null @@ -1,16 +0,0 @@ - add_float.py:f - -input1 -input2output"AddaddZ -input1 - - -Z -input2 - - -b -output - - -B
\ No newline at end of file diff --git a/model-integration/src/test/models/onnx/add_float.py b/model-integration/src/test/models/onnx/add_float.py deleted file mode 100755 index e18b2c46d9d..00000000000 --- a/model-integration/src/test/models/onnx/add_float.py +++ /dev/null @@ -1,27 +0,0 @@ -# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import onnx -from onnx import helper, TensorProto - -INPUT_1 = helper.make_tensor_value_info('input1', TensorProto.FLOAT, [1]) -INPUT_2 = helper.make_tensor_value_info('input2', TensorProto.FLOAT, [1]) -OUTPUT = helper.make_tensor_value_info('output', TensorProto.FLOAT, [1]) - -nodes = [ - helper.make_node( - 'Add', - ['input1', 'input2'], - ['output'], - ), -] -graph_def = helper.make_graph( - nodes, - 'add', - [ - INPUT_1, - INPUT_2 - ], - [OUTPUT], -) -model_def = helper.make_model(graph_def, producer_name='add_float.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) -onnx.save(model_def, 'add_float.onnx') diff --git a/model-integration/src/test/models/onnx/add_int64.onnx b/model-integration/src/test/models/onnx/add_int64.onnx deleted file mode 100644 index 7b3a9ec6b95..00000000000 --- a/model-integration/src/test/models/onnx/add_int64.onnx +++ /dev/null @@ -1,16 +0,0 @@ - add_int64.py:f - -input1 -input2output"AddaddZ -input1 - - -Z -input2 - - -b -output - - -B
\ No newline at end of file diff --git a/model-integration/src/test/models/onnx/add_int64.py b/model-integration/src/test/models/onnx/add_int64.py deleted file mode 100755 index 87908e292a2..00000000000 --- a/model-integration/src/test/models/onnx/add_int64.py +++ /dev/null @@ -1,27 +0,0 @@ -# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import onnx -from onnx import helper, TensorProto - -INPUT_1 = helper.make_tensor_value_info('input1', TensorProto.INT64, [1]) -INPUT_2 = helper.make_tensor_value_info('input2', TensorProto.INT64, [1]) -OUTPUT = helper.make_tensor_value_info('output', TensorProto.INT64, [1]) - -nodes = [ - helper.make_node( - 'Add', - ['input1', 'input2'], - ['output'], - ), -] -graph_def = helper.make_graph( - nodes, - 'add', - [ - INPUT_1, - INPUT_2 - ], - [OUTPUT], -) -model_def = helper.make_model(graph_def, producer_name='add_int64.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) -onnx.save(model_def, 'add_int64.onnx') diff --git a/model-integration/src/test/models/onnx/cast_bfloat16_float.onnx b/model-integration/src/test/models/onnx/cast_bfloat16_float.onnx deleted file mode 100644 index cb19592abf4..00000000000 --- a/model-integration/src/test/models/onnx/cast_bfloat16_float.onnx +++ /dev/null @@ -1,12 +0,0 @@ -cast_bfloat16_float.py:U -! -input1output"Cast* -to castZ -input1 - - -b -output - - -B
\ No newline at end of file diff --git a/model-integration/src/test/models/onnx/cast_bfloat16_float.py b/model-integration/src/test/models/onnx/cast_bfloat16_float.py deleted file mode 100755 index 14b05347262..00000000000 --- a/model-integration/src/test/models/onnx/cast_bfloat16_float.py +++ /dev/null @@ -1,24 +0,0 @@ -# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import onnx -from onnx import helper, TensorProto - -INPUT_1 = helper.make_tensor_value_info('input1', TensorProto.BFLOAT16, [1]) -OUTPUT = helper.make_tensor_value_info('output', TensorProto.FLOAT, [1]) - -nodes = [ - helper.make_node( - 'Cast', - ['input1'], - ['output'], - to=TensorProto.FLOAT - ), -] -graph_def = helper.make_graph( - nodes, - 'cast', - [INPUT_1], - [OUTPUT], -) -model_def = helper.make_model(graph_def, producer_name='cast_bfloat16_float.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) -onnx.save(model_def, 'cast_bfloat16_float.onnx') diff --git a/model-integration/src/test/models/onnx/cast_float_int8.onnx b/model-integration/src/test/models/onnx/cast_float_int8.onnx deleted file mode 100644 index c30b023dd68..00000000000 --- a/model-integration/src/test/models/onnx/cast_float_int8.onnx +++ /dev/null @@ -1,12 +0,0 @@ -cast_float_int8.py:U -! -input1output"Cast* -to castZ -input1 - - -b -output - - -B
\ No newline at end of file diff --git a/model-integration/src/test/models/onnx/cast_float_int8.py b/model-integration/src/test/models/onnx/cast_float_int8.py deleted file mode 100755 index bdc0850d033..00000000000 --- a/model-integration/src/test/models/onnx/cast_float_int8.py +++ /dev/null @@ -1,24 +0,0 @@ -# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import onnx -from onnx import helper, TensorProto - -INPUT_1 = helper.make_tensor_value_info('input1', TensorProto.FLOAT, [1]) -OUTPUT = helper.make_tensor_value_info('output', TensorProto.INT8, [1]) - -nodes = [ - helper.make_node( - 'Cast', - ['input1'], - ['output'], - to=TensorProto.INT8 - ), -] -graph_def = helper.make_graph( - nodes, - 'cast', - [INPUT_1], - [OUTPUT], -) -model_def = helper.make_model(graph_def, producer_name='cast_float_int8.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) -onnx.save(model_def, 'cast_float_int8.onnx') diff --git a/model-integration/src/test/models/onnx/cast_int8_float.onnx b/model-integration/src/test/models/onnx/cast_int8_float.onnx deleted file mode 100644 index 65aea4a36ae..00000000000 --- a/model-integration/src/test/models/onnx/cast_int8_float.onnx +++ /dev/null @@ -1,12 +0,0 @@ -cast_int8_float.py:U -! -input1output"Cast* -to castZ -input1 - - -b -output - - -B
\ No newline at end of file diff --git a/model-integration/src/test/models/onnx/cast_int8_float.py b/model-integration/src/test/models/onnx/cast_int8_float.py deleted file mode 100755 index 70bf2cf70ca..00000000000 --- a/model-integration/src/test/models/onnx/cast_int8_float.py +++ /dev/null @@ -1,24 +0,0 @@ -# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import onnx -from onnx import helper, TensorProto - -INPUT_1 = helper.make_tensor_value_info('input1', TensorProto.INT8, [1]) -OUTPUT = helper.make_tensor_value_info('output', TensorProto.FLOAT, [1]) - -nodes = [ - helper.make_node( - 'Cast', - ['input1'], - ['output'], - to=TensorProto.FLOAT - ), -] -graph_def = helper.make_graph( - nodes, - 'cast', - [INPUT_1], - [OUTPUT], -) -model_def = helper.make_model(graph_def, producer_name='cast_int8_float.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) -onnx.save(model_def, 'cast_int8_float.onnx') diff --git a/model-integration/src/test/models/onnx/pytorch/one_layer.onnx b/model-integration/src/test/models/onnx/pytorch/one_layer.onnx Binary files differdeleted file mode 100644 index dc9f664b943..00000000000 --- a/model-integration/src/test/models/onnx/pytorch/one_layer.onnx +++ /dev/null diff --git a/model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py b/model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py deleted file mode 100755 index 1296d84e180..00000000000 --- a/model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py +++ /dev/null @@ -1,38 +0,0 @@ -# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import torch -import torch.onnx - - -class MyModel(torch.nn.Module): - def __init__(self): - super(MyModel, self).__init__() - self.linear = torch.nn.Linear(in_features=3, out_features=1) - self.logistic = torch.nn.Sigmoid() - - def forward(self, vec): - return self.logistic(self.linear(vec)) - - -def main(): - model = MyModel() - - # Omit training - just export randomly initialized network - - data = torch.FloatTensor([[0.1, 0.2, 0.3],[0.4, 0.5, 0.6]]) - torch.onnx.export(model, - data, - "one_layer.onnx", - input_names = ["input"], - output_names = ["output"], - dynamic_axes = { - "input": {0: "batch"}, - "output": {0: "batch"}, - }, - opset_version=12) - - -if __name__ == "__main__": - main() - - diff --git a/model-integration/src/test/models/onnx/simple/matmul.onnx b/model-integration/src/test/models/onnx/simple/matmul.onnx deleted file mode 100644 index 9bb88406116..00000000000 --- a/model-integration/src/test/models/onnx/simple/matmul.onnx +++ /dev/null @@ -1,16 +0,0 @@ - matmul.py:x - -input1 -input2output"MatMulmatmulZ -input1 - - -Z -input2 - - -b -output - - -B
\ No newline at end of file diff --git a/model-integration/src/test/models/onnx/simple/matmul.py b/model-integration/src/test/models/onnx/simple/matmul.py deleted file mode 100755 index beec55e9f5a..00000000000 --- a/model-integration/src/test/models/onnx/simple/matmul.py +++ /dev/null @@ -1,27 +0,0 @@ -# Copyright 2020 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. - -import onnx -from onnx import helper, TensorProto - -INPUT1 = helper.make_tensor_value_info('input1', TensorProto.FLOAT, [2, 3]) -INPUT2 = helper.make_tensor_value_info('input2', TensorProto.FLOAT, [3, 4]) -OUTPUT = helper.make_tensor_value_info('output', TensorProto.FLOAT, [2, 4]) - -nodes = [ - helper.make_node( - 'MatMul', - ['input1', 'input2'], - ['output'], - ), -] -graph_def = helper.make_graph( - nodes, - 'matmul', - [ - INPUT1, - INPUT2, - ], - [OUTPUT], -) -model_def = helper.make_model(graph_def, producer_name='matmul.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) -onnx.save(model_def, 'matmul.onnx') |