Searched refs:NNAPI_FuseCode (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | serializer.py | 135 class NNAPI_FuseCode: class
855 node, NNAPI_OperationCode.ADD, NNAPI_FuseCode.FUSED_NONE
858 node, NNAPI_OperationCode.SUB, NNAPI_FuseCode.FUSED_NONE
861 node, NNAPI_OperationCode.MUL, NNAPI_FuseCode.FUSED_NONE
864 node, NNAPI_OperationCode.DIV, NNAPI_FuseCode.FUSED_NONE
892 node, NNAPI_FuseCode.FUSED_NONE
895 node, NNAPI_FuseCode.FUSED_RELU
898 node, NNAPI_FuseCode.FUSED_NONE, transpose=True
901 node, NNAPI_OperationCode.ADD, NNAPI_FuseCode.FUSED_NONE
904 node, NNAPI_OperationCode.ADD, NNAPI_FuseCode.FUSED_RELU
[all …]
|