{ "cases": [ { "name": "small_64_boxes", "inputs": { "boxes": { "dtype": "float32", "shape": [1, 64, 4], "data": { "kind": "constant", "value": 0.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 64], "data": { "kind": "constant", "value": 1.0 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 10 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.1 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [10, 3] } }, "bench": { "metrics": [ { "type": "gflops", "value": "7 * dim(shapes.boxes, 0) * dim(shapes.scores, 1) * dim(shapes.boxes, 1) * dim(shapes.boxes, 1)" } ] } }, { "name": "nms-1class-2048-boxes", "preset": "smoke", "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2048, 4], "data": { "kind": "constant", "value": 0.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 2048], "data": { "kind": "constant", "value": 1.0 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 64 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [64, 3] } }, "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms-8class-1024-boxes", "preset": "smoke", "inputs": { "boxes": { "dtype": "float32", "shape": [1, 1024, 4], "data": { "kind": "constant", "value": 0.0 } }, "scores": { "dtype": "float32", "shape": [1, 8, 1024], "data": { "kind": "constant", "value": 1.0 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [256, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_single_class_4096_boxes_max_keep_wg_parallel", "preset": "stress", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 4096, 4], "data": { "kind": "constant", "value": 0.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 4096], "data": { "kind": "constant", "value": 1.0 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 4096 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.01 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [4096, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "yolo-80class-8400-boxes-top100", "preset": "stress", "provenance": { "notes": "Worst-case exact-output workload: zero-area boxes never overlap and unit scores pass the threshold, so every class reaches max_output_boxes_per_class." }, "attrs": { "center_point_box": 1 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8400, 4], "data": { "kind": "constant", "value": 0.0 } }, "scores": { "dtype": "float32", "shape": [1, 80, 8400], "data": { "kind": "constant", "value": 1.0 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.25 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [8000, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n255_wg256", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 255, 4], "data": { "kind": "linspace", "start": 0.0, "end": 1019.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 255], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n256_wg256", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 256, 4], "data": { "kind": "linspace", "start": 0.0, "end": 1023.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 256], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n257_wg256", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 257, 4], "data": { "kind": "linspace", "start": 0.0, "end": 1027.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 257], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n8192_wg256", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8192, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32767.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 8192], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n8193_wg256", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8193, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32771.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 8193], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n32768_wg256", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 32768, 4], "data": { "kind": "linspace", "start": 0.0, "end": 131071.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 32768], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n32769_wg256", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 32769, 4], "data": { "kind": "linspace", "start": 0.0, "end": 131075.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 32769], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "private_flags_n1025_wg128", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 1025, 4], "data": { "kind": "linspace", "start": 0.0, "end": 4099.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 1025], "data": { "kind": "cycle", "values": [0.9, 0.25, -0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.45] } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [0.25] } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [15, 3] } }, "provenance": { "notes": "Non-overlapping positive-area boxes and tied scores above, equal to, and below the threshold. Five independent classes check exact score filtering and class-local output limits." }, "tunables": { "WORKGROUP_SIZE": 128 }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms-multiclass-sparse-scores", "preset": "stress", "provenance": { "notes": "Separated positive-area boxes never overlap. One score per sixteen candidates exceeds the threshold; every class still supplies at least one hundred candidates. The selected indices have an independent exact arithmetic reference." }, "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8400, 4], "data": { "kind": "linspace", "start": 0.0, "end": 33599.0 } }, "scores": { "dtype": "float32", "shape": [1, 80, 8400], "data": { "kind": "cycle", "values": [0.9, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.25 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [8000, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n1024_c8_k32_ascending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 1024, 4], "data": { "kind": "linspace", "start": 0.0, "end": 4095.0 } }, "scores": { "dtype": "float32", "shape": [1, 8, 1024], "data": { "kind": "linspace", "start": -0.9, "end": 0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [256, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n1024_c8_k32_descending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 1024, 4], "data": { "kind": "linspace", "start": 0.0, "end": 4095.0 } }, "scores": { "dtype": "float32", "shape": [1, 8, 1024], "data": { "kind": "linspace", "start": 0.9, "end": -0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [256, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n1024_c8_k32_mixed", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 1024, 4], "data": { "kind": "linspace", "start": 0.0, "end": 4095.0 } }, "scores": { "dtype": "float32", "shape": [1, 8, 1024], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [256, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n2048_c1_k64_ascending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2048, 4], "data": { "kind": "linspace", "start": 0.0, "end": 8191.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 2048], "data": { "kind": "linspace", "start": -0.9, "end": 0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 64 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [64, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n2048_c1_k64_descending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2048, 4], "data": { "kind": "linspace", "start": 0.0, "end": 8191.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 2048], "data": { "kind": "linspace", "start": 0.9, "end": -0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 64 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [64, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n2048_c1_k64_mixed", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2048, 4], "data": { "kind": "linspace", "start": 0.0, "end": 8191.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 2048], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 64 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [64, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n2051_c5_k100_ascending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2051, 4], "data": { "kind": "linspace", "start": 0.0, "end": 8203.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 2051], "data": { "kind": "linspace", "start": -0.9, "end": 0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [500, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n2051_c5_k100_descending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2051, 4], "data": { "kind": "linspace", "start": 0.0, "end": 8203.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 2051], "data": { "kind": "linspace", "start": 0.9, "end": -0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [500, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n2051_c5_k100_mixed", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2051, 4], "data": { "kind": "linspace", "start": 0.0, "end": 8203.0 } }, "scores": { "dtype": "float32", "shape": [1, 5, 2051], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [500, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n4096_c16_k100_ascending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 4096, 4], "data": { "kind": "linspace", "start": 0.0, "end": 16383.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 4096], "data": { "kind": "linspace", "start": -0.9, "end": 0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n4096_c16_k100_descending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 4096, 4], "data": { "kind": "linspace", "start": 0.0, "end": 16383.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 4096], "data": { "kind": "linspace", "start": 0.9, "end": -0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n4096_c16_k100_mixed", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 4096, 4], "data": { "kind": "linspace", "start": 0.0, "end": 16383.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 4096], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8192_c16_k100_ascending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8192, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32767.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 8192], "data": { "kind": "linspace", "start": -0.9, "end": 0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8192_c16_k100_descending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8192, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32767.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 8192], "data": { "kind": "linspace", "start": 0.9, "end": -0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8192_c16_k100_mixed", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8192, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32767.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 8192], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8193_c16_k100_ascending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8193, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32771.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 8193], "data": { "kind": "linspace", "start": -0.9, "end": 0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8193_c16_k100_descending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8193, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32771.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 8193], "data": { "kind": "linspace", "start": 0.9, "end": -0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8193_c16_k100_mixed", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8193, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32771.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 8193], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8400_c80_k100_ascending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8400, 4], "data": { "kind": "linspace", "start": 0.0, "end": 33599.0 } }, "scores": { "dtype": "float32", "shape": [1, 80, 8400], "data": { "kind": "linspace", "start": -0.9, "end": 0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [8000, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8400_c80_k100_descending", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8400, 4], "data": { "kind": "linspace", "start": 0.0, "end": 33599.0 } }, "scores": { "dtype": "float32", "shape": [1, 80, 8400], "data": { "kind": "linspace", "start": 0.9, "end": -0.9 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [8000, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8400_c80_k100_mixed", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8400, 4], "data": { "kind": "linspace", "start": 0.0, "end": 33599.0 } }, "scores": { "dtype": "float32", "shape": [1, 80, 8400], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": -2.0 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [8000, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n2048_c1_k64_mixed_filtered", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 2048, 4], "data": { "kind": "linspace", "start": 0.0, "end": 8191.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 2048], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 64 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.6 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [64, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n4096_c16_k100_mixed_filtered", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 4096, 4], "data": { "kind": "linspace", "start": 0.0, "end": 16383.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 4096], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.6 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8193_c16_k100_mixed_filtered", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8193, 4], "data": { "kind": "linspace", "start": 0.0, "end": 32771.0 } }, "scores": { "dtype": "float32", "shape": [1, 16, 8193], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.6 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [1600, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "nms_neighbor_n8400_c80_k100_mixed_filtered", "attrs": { "center_point_box": 0 }, "inputs": { "boxes": { "dtype": "float32", "shape": [1, 8400, 4], "data": { "kind": "linspace", "start": 0.0, "end": 33599.0 } }, "scores": { "dtype": "float32", "shape": [1, 80, 8400], "data": { "kind": "fillFloat32", "sinStep": 0.713, "cosStep": 0.193, "scale": 0.7 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 100 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.45 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.6 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [8000, 3] } }, "bench": { "metrics": [{ "type": "bandwidth", "value": "4 * (numel(shapes.boxes) + numel(shapes.scores))" }] } }, { "name": "serial_control_n257_k32_ties", "inputs": { "boxes": { "dtype": "float32", "shape": [1, 257, 4], "data": { "kind": "linspace", "start": 0.0, "end": 1027.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 257], "data": { "kind": "constant", "value": 1.0 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.1 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [32, 3] } }, "bench": { "metrics": [ { "type": "gflops", "value": "7 * dim(shapes.boxes, 0) * dim(shapes.scores, 1) * dim(shapes.boxes, 1) * dim(shapes.boxes, 1)" } ] } }, { "name": "serial_control_n257_k32_ascending", "inputs": { "boxes": { "dtype": "float32", "shape": [1, 257, 4], "data": { "kind": "linspace", "start": 0.0, "end": 1027.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 257], "data": { "kind": "linspace", "start": 0.2, "end": 1.0 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.1 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [32, 3] } }, "bench": { "metrics": [ { "type": "gflops", "value": "7 * dim(shapes.boxes, 0) * dim(shapes.scores, 1) * dim(shapes.boxes, 1) * dim(shapes.boxes, 1)" } ] } }, { "name": "serial_control_n257_k32_descending", "inputs": { "boxes": { "dtype": "float32", "shape": [1, 257, 4], "data": { "kind": "linspace", "start": 0.0, "end": 1027.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 257], "data": { "kind": "linspace", "start": 1.0, "end": 0.2 } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.1 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [32, 3] } }, "bench": { "metrics": [ { "type": "gflops", "value": "7 * dim(shapes.boxes, 0) * dim(shapes.scores, 1) * dim(shapes.boxes, 1) * dim(shapes.boxes, 1)" } ] } }, { "name": "serial_control_n257_k32_mixed", "inputs": { "boxes": { "dtype": "float32", "shape": [1, 257, 4], "data": { "kind": "linspace", "start": 0.0, "end": 1027.0 } }, "scores": { "dtype": "float32", "shape": [1, 1, 257], "data": { "kind": "cycle", "values": [0.9, 0.25, 0.1, 0.7, 0.3, 0.85, 0.55] } }, "max_output_boxes_per_class": { "dtype": "uint32", "shape": [], "data": { "kind": "constant", "value": 32 } }, "iou_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.5 } }, "score_threshold": { "dtype": "float32", "shape": [], "data": { "kind": "constant", "value": 0.1 } } }, "outputs": { "selected_indices": { "dtype": "uint32", "shape": [32, 3] } }, "bench": { "metrics": [ { "type": "gflops", "value": "7 * dim(shapes.boxes, 0) * dim(shapes.scores, 1) * dim(shapes.boxes, 1) * dim(shapes.boxes, 1)" } ] } } ] }