Searched refs:NNAPI_OperationCode (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | serializer.py | 37 class NNAPI_OperationCode: class
624 self.add_operation(NNAPI_OperationCode.TRANSPOSE, inputs, outputs)
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
867 node, NNAPI_OperationCode.RELU
870 node, NNAPI_OperationCode.LOGISTIC
876 node, NNAPI_OperationCode.MAX_POOL_2D
901 node, NNAPI_OperationCode.ADD, NNAPI_FuseCode.FUSED_NONE
[all …]
|