33 lines
1.5 KiB
Modula-2
33 lines
1.5 KiB
Modula-2
BEGINE_DEFINE_TARGET(neutral)
|
|
DEFINE_RUNTIME_OP(neutral, binary, Binary, 0)
|
|
DEFINE_RUNTIME_OP(neutral, concat, Concat, 1)
|
|
DEFINE_RUNTIME_OP(neutral, conv2d, Conv2D, 2)
|
|
DEFINE_RUNTIME_OP(neutral, dequantize, Dequantize, 3)
|
|
DEFINE_RUNTIME_OP(neutral, matmul, MatMul, 4)
|
|
DEFINE_RUNTIME_OP(neutral, pad, Pad, 5)
|
|
DEFINE_RUNTIME_OP(neutral, quantize, Quantize, 6)
|
|
DEFINE_RUNTIME_OP(neutral, reduce, Reduce, 7)
|
|
DEFINE_RUNTIME_OP(neutral, reduce_window2d, ReduceWindow2D, 8)
|
|
DEFINE_RUNTIME_OP(neutral, memory_copy, MemoryCopy, 9)
|
|
DEFINE_RUNTIME_OP(neutral, resize_bilinear, ResizeBilinear, 10)
|
|
DEFINE_RUNTIME_OP(neutral, resize_nearest_neighbor, ResizeNearestNeighbor, 11)
|
|
DEFINE_RUNTIME_OP(neutral, softmax, Softmax, 12)
|
|
DEFINE_RUNTIME_OP(neutral, transpose, Transpose, 13)
|
|
DEFINE_RUNTIME_OP(neutral, strided_slice, StridedSlice, 14)
|
|
END_DEFINE_TARGET()
|
|
|
|
// CPU
|
|
BEGINE_DEFINE_TARGET(cpu)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_conv2d, CPU_CPUConv2D, 1001)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_depthwise_conv2d, CPU_CPUDepthwiseConv2D, 1002)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_reduce_window2d, CPU_CPUReduceWindow2D, 1003)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_quantized_conv2d, CPU_CPUQuantizedConv2D, 1004)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_quantized_depthwise_conv2d, CPU_CPUQuantizedDepthwiseConv2D, 1005)
|
|
END_DEFINE_TARGET()
|
|
|
|
// K210
|
|
BEGINE_DEFINE_TARGET(k210)
|
|
DEFINE_RUNTIME_OP(k210, kpu_upload, K210_KPUUpload, 2001)
|
|
DEFINE_RUNTIME_OP(k210, kpu_conv2d, K210_KPUConv2D, 2002)
|
|
END_DEFINE_TARGET()
|