NNAPI conversion for a CNN - node kind is not in ADDER_MAP

ekremcet · February 15, 2021, 9:57am

Some layers are not supported by NNAPI system yet. You can see the list of supported ops here:

pytorch/pytorch/blob/master/torch/backends/_nnapi/serializer.py#L585-L636


ADDER_MAP = {
    "prim::GetAttr": lambda self, node:
        self.add_getattr(node),
    "prim::Constant": lambda self, node:
        self.add_constant_node(node),
    "prim::ListConstruct": lambda self, node:
        self.add_list_construct(node),
    "prim::TupleConstruct": lambda self, node:
        self.add_tuple_construct(node),
    "aten::reshape": lambda self, node:
        self.add_reshape(node),
    "aten::size": lambda self, node:
        self.add_size(node),
    "aten::quantize_per_tensor": lambda self, node:
        self.add_quantize(node),
    "aten::dequantize": lambda self, node:
        self.add_dequantize(node),
    "aten::add": lambda self, node:
        self.add_add_sub_op(node, NNAPI_OperationCode.ADD, NNAPI_FuseCode.FUSED_NONE),
    "aten::sub": lambda self, node:

This file has been truncated. show original