From a21b2e88d19d8cb11a9120d40bacbb594d600565 Mon Sep 17 00:00:00 2001 From: Won Jeon Date: Thu, 10 Aug 2023 10:33:01 +0000 Subject: Add DIM operator to reference model Signed-off-by: Won Jeon Change-Id: Iea11ee5d3d98773e9c5e9b827593c05afb41ce3b --- verif/conformance/test_select.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'verif/conformance/test_select.py') diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py index f4e2a61..b7bbfc3 100644 --- a/verif/conformance/test_select.py +++ b/verif/conformance/test_select.py @@ -506,6 +506,13 @@ class DepthwiseConv2dOperator(Operator): param_names = ["kernel", "shape", "type", "accum_type", "stride", "pad", "dilation"] +class DimOeprator(Operator): + """Test selector for the DIM operator.""" + + name = "dim" + param_names = ["shape", "type", "axis"] + + class EqualOperator(Operator): """Test selector for the EQUAL operator.""" -- cgit v1.2.1