Imported Upstream version 1.9.0
[platform/core/ml/nnfw.git] / res / TensorFlowLiteRecipes / NonMaxSuppressionV5_000 / test.recipe
1 operand {
2   name: "boxes"
3   type: FLOAT32
4   shape { dim: 10 dim: 4 }
5 }
6 operand {
7   name: "scores"
8   type: FLOAT32
9   shape { dim: 10 }
10 }
11 operand {
12   name: "max_output_size"
13   type: INT32
14   shape { }
15 }
16 operand {
17   name: "iou_threshold"
18   type: FLOAT32
19   shape { }
20 }
21 operand {
22   name: "score_threshold"
23   type: FLOAT32
24   shape { }
25 }
26 operand {
27   name: "soft_nms_sigma"
28   type: FLOAT32
29   shape { }
30 }
31 operand {
32   name: "selected_indices"
33   type: INT32
34   shape { }
35 }
36 operand {
37   name: "selected_scores"
38   type: FLOAT32
39   shape { }
40 }
41 operand {
42   name: "valid_outputs"
43   type: INT32
44   shape { }
45 }
46 operation {
47   type: "NonMaxSuppressionV5"
48   input: "boxes"
49   input: "scores"
50   input: "max_output_size"
51   input: "iou_threshold"
52   input: "score_threshold"
53   input: "soft_nms_sigma"
54   output: "selected_indices"
55   output: "selected_scores"
56   output: "valid_outputs"
57 }
58 input: "boxes"
59 input: "scores"
60 input: "max_output_size"
61 input: "iou_threshold"
62 input: "score_threshold"
63 input: "soft_nms_sigma"
64 output: "selected_indices"
65 output: "selected_scores"
66 output: "valid_outputs"