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 | 36 |
1 files changed, 5 insertions, 31 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 6e9ba356293..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,25 +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 - } - 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 } } @@ -55,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") } } @@ -68,29 +62,9 @@ search test { } summary-features { onnxModel(another_model).out - onnxModel("files/summary_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} + onnxModel("files/ranking_model.onnx", "path/to/output:2") } - } - 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} - } } } |