TIM-VX/include/tim/vx/ops/MaxpoolWithArgmax2.json at 6a37d38f8b396eebfb96700e98119da241ec200f · chenfeiyue-cfy/TIM-VX · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
{
    "MaxpoolWithArgmax2":{
        "parameters":
            [
                {"name": "padding",
                    "dtype": "tim::vx::PadType",
                    "range": ["tim::vx::PadType::NONE",
                              "tim::vx::PadType::AUTO",
                              "tim::vx::PadType::VALID",
                              "tim::vx::PadType::SAME"]
                },
                {"name": "ksize",
                    "dtype": "std::array<uint32_t,2>",
                    "Optional": "true",
                    "default": "{0,0}"
                },
                {"name": "stride",
                    "dtype": "std::array<uint32_t,2>"
                },
                {"name": "round_type",
                "dtype": "tim::vx::RoundType",
                "Optional": "true",
                "default": "tim::vx::RoundType::FLOOR",
                "range": ["tim::vx::RoundType::FLOOR",
                          "tim::vx::RoundType::CEILING"]
                },
                {"name": "layout",
                    "dtype": "tim::vx::DataLayout",
                    "Optional": "true",
                    "default": "tim::vx::DataLayout::WHCN",
                    "range":["tim::vx::DataLayout::ANY",
                             "tim::vx::DataLayout::WHCN",
                             "tim::vx::DataLayout::CWHN"]
                }

            ]
    }
}