summaryrefslogtreecommitdiffstats
path: root/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd
diff options
context:
space:
mode:
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.sd36
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}
- }
}
}