diff options
Diffstat (limited to 'config-model/src/test/cfg/application')
3 files changed, 76 insertions, 0 deletions
diff --git a/config-model/src/test/cfg/application/onnx_cluster_specific/models/mul.onnx b/config-model/src/test/cfg/application/onnx_cluster_specific/models/mul.onnx new file mode 100644 index 00000000000..087e2c3427f --- /dev/null +++ b/config-model/src/test/cfg/application/onnx_cluster_specific/models/mul.onnx @@ -0,0 +1,16 @@ +mul.py:f + +input1 +input2output"MulmulZ +input1 + + +Z +input2 + + +b +output + + +B
\ No newline at end of file diff --git a/config-model/src/test/cfg/application/onnx_cluster_specific/models/mul.py b/config-model/src/test/cfg/application/onnx_cluster_specific/models/mul.py new file mode 100755 index 00000000000..9fcb8612af9 --- /dev/null +++ b/config-model/src/test/cfg/application/onnx_cluster_specific/models/mul.py @@ -0,0 +1,26 @@ +# Copyright Yahoo. 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( + 'Mul', + ['input1', 'input2'], + ['output'], + ), +] +graph_def = helper.make_graph( + nodes, + 'mul', + [ + INPUT_1, + INPUT_2 + ], + [OUTPUT], +) +model_def = helper.make_model(graph_def, producer_name='mul.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) +onnx.save(model_def, 'mul.onnx') diff --git a/config-model/src/test/cfg/application/onnx_cluster_specific/services.xml b/config-model/src/test/cfg/application/onnx_cluster_specific/services.xml new file mode 100644 index 00000000000..06b9a8c3a55 --- /dev/null +++ b/config-model/src/test/cfg/application/onnx_cluster_specific/services.xml @@ -0,0 +1,34 @@ +<?xml version="1.0" encoding="utf-8" ?> +<!-- Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. --> +<services version="1.0"> + + <container id="c1" version="1.0"> + <model-evaluation> + <onnx> + <models> + <model name="mul"> + <intraop-threads>2</intraop-threads> + <gpu-device>0</gpu-device> + </model> + </models> + </onnx> + </model-evaluation> + </container> + + <container id="c2" version="1.0"> + <http> + <server id="c1Server" port="8081" /> + </http> + <model-evaluation> + <onnx> + <models> + <model name="mul"> + <intraop-threads>4</intraop-threads> + <gpu-device>1</gpu-device> + </model> + </models> + </onnx> + </model-evaluation> + </container> + +</services> |