Searched refs:DepthToSpaceCPUKernel (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | depthtospace_cpu_kernel.h | 28 class DepthToSpaceCPUKernel : public CPUKernel {
30 DepthToSpaceCPUKernel() = default;
31 ~DepthToSpaceCPUKernel() override = default;
45 DepthToSpaceCPUKernel, float);
49 DepthToSpaceCPUKernel, float16);
53 DepthToSpaceCPUKernel, int8_t);
57 DepthToSpaceCPUKernel, int16_t);
61 DepthToSpaceCPUKernel, int);
65 DepthToSpaceCPUKernel, int64_t);
69 DepthToSpaceCPUKernel, uint8_t);
[all …]
|
| D | depthtospace_cpu_kernel.cc | 27 void DepthToSpaceCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel()
36 bool DepthToSpaceCPUKernel<T>::Launch(const std::vector<kernel::AddressPtr> &inputs, in Launch()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | depth_to_space_fp32.cc | 30 int DepthToSpaceCPUKernel::Init() { in Init()
40 int DepthToSpaceCPUKernel::ReSize() { in ReSize()
67 int DepthToSpaceCPUKernel::Run() { in Run()
83 …NEL(kCPU, kNumberTypeFloat32, PrimitiveType_DepthToSpace, LiteKernelCreator<DepthToSpaceCPUKernel>)
|
| D | depth_to_space_fp32.h | 24 class DepthToSpaceCPUKernel : public InnerKernel {
26 DepthToSpaceCPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in DepthToSpaceCPUKernel() function
31 ~DepthToSpaceCPUKernel() = default;
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | depth_to_space_fp16.h | 23 class DepthToSpaceFp16CPUKernel : public DepthToSpaceCPUKernel {
27 : DepthToSpaceCPUKernel(parameter, inputs, outputs, ctx) {} in DepthToSpaceFp16CPUKernel()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | depth_to_space_int8.h | 27 class DepthToSpaceInt8CPUKernel : public DepthToSpaceCPUKernel {
31 : DepthToSpaceCPUKernel(parameter, inputs, outputs, ctx) {} in DepthToSpaceInt8CPUKernel()
|