33 lines
1.8 KiB
Modula-2
33 lines
1.8 KiB
Modula-2
BEGINE_DEFINE_TARGET(neutral)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(binary, Binary, 0x0)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(concat, Concat, 0x1)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(conv2d, Conv2D, 0x2)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(dequantize, Dequantize, 0x3)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(matmul, MatMul, 0x4)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(pad, Pad, 0x5)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(quantize, Quantize, 0x6)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(reduce, Reduce, 0x7)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(reduce_window2d, ReduceWindow2D, 0x8)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(memory_copy, MemoryCopy, 0x9)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(resize_image, ResizeImage, 0x0A)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(softmax, Softmax, 0x0B)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(transpose, Transpose, 0x0C)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(strided_slice, StridedSlice, 0x0D)
|
|
DEFINE_NEUTRAL_RUNTIME_OP(unary, Unary, 0x0E)
|
|
END_DEFINE_TARGET()
|
|
|
|
// CPU
|
|
BEGINE_DEFINE_TARGET(cpu)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_conv2d, CPUConv2D, 0x1001)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_depthwise_conv2d, CPUDepthwiseConv2D, 0x1002)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_reduce_window2d, CPUReduceWindow2D, 0x1003)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_quantized_conv2d, CPUQuantizedConv2D, 0x1004)
|
|
DEFINE_RUNTIME_OP(cpu, cpu_quantized_depthwise_conv2d, CPUQuantizedDepthwiseConv2D, 0x1005)
|
|
END_DEFINE_TARGET()
|
|
|
|
// K210
|
|
BEGINE_DEFINE_TARGET(k210)
|
|
DEFINE_RUNTIME_OP(k210, kpu_upload, KPUUpload, 0x2001)
|
|
DEFINE_RUNTIME_OP(k210, kpu_conv2d, KPUConv2D, 0x2002)
|
|
END_DEFINE_TARGET()
|