diff options
Diffstat (limited to 'config-model/src/test/integration/onnx-model/searchdefinitions/test.sd')
-rw-r--r-- | config-model/src/test/integration/onnx-model/searchdefinitions/test.sd | 52 |
1 files changed, 5 insertions, 47 deletions
diff --git a/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd b/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd index a87222e77ee..0f0fa694e6f 100644 --- a/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd +++ b/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd @@ -14,7 +14,7 @@ search test { } onnx-model my_model { - file: files/model.onnx + file: files/ranking_model.onnx input first_input: attribute(document_field) input "second/input:0": constant(my_constant) input "third_input": my_function @@ -22,31 +22,19 @@ search test { } onnx-model another_model { - file: files/model.onnx + file: files/ranking_model.onnx input first_input: attribute(document_field) input "second/input:0": constant(my_constant) input "third_input": another_function output "path/to/output:2": out } - onnx-model dynamic_model { - file: files/dynamic_model.onnx - input input: my_function - output output: my_output - } - - onnx-model unbound_model { - file: files/unbound_model.onnx - input input: my_function - output output: my_output - } - rank-profile test_model_config { function my_function() { expression: tensor(d0[2])(1) } first-phase { - expression: onnxModel(my_model).out{d0:1} + expression: onnxModel(my_model).out } } @@ -61,7 +49,7 @@ search test { expression: my_function() } first-phase { - expression: onnxModel("files/model.onnx", "path/to/output:1"){d0:1} + expression: onnxModel("files/ranking_model.onnx", "path/to/output:1") } } @@ -74,39 +62,9 @@ search test { } summary-features { onnxModel(another_model).out - onnxModel("files/summary_model.onnx", "path/to/output:2") + onnxModel("files/ranking_model.onnx", "path/to/output:2") } - } - rank-profile test_dynamic_model { - function my_function() { - expression: tensor(d0[1],d1[2])(d1) - } - first-phase { - expression: onnxModel(dynamic_model){d0:0,d1:1} - } } - rank-profile test_dynamic_model_2 { - function my_function_2() { - expression: tensor(d0[1],d1[3])(d1) - } - function my_function() { - expression: my_function_2() - } - first-phase { - expression: onnxModel(dynamic_model){d0:0,d1:2} - } - } - - rank-profile test_unbound_model { - function my_function() { - expression: tensor(d0[1],d1[2])(d1) - } - first-phase { - expression: onnxModel(unbound_model){d0:0,d1:1} - } - } - - } |