Browse Source

!4659 not reg fp16 transpose mean concat

Merge pull request !4659 from zhaozhenlong/lite/issue/no_reg_fp16_transpose_mean_concat
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
6e8866117f
3 changed files with 4 additions and 4 deletions
  1. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp16/concat_fp16.cc
  2. +2
    -2
      mindspore/lite/src/runtime/kernel/arm/fp16/reduce_fp16.cc
  3. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp16/transpose_fp16.cc

+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp16/concat_fp16.cc View File

@@ -164,5 +164,5 @@ kernel::LiteKernel *CpuConcatFp16KernelCreator(const std::vector<lite::tensor::T
}
return kernel;
}
REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Concat, CpuConcatFp16KernelCreator)
// REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Concat, CpuConcatFp16KernelCreator)
} // namespace mindspore::kernel

+ 2
- 2
mindspore/lite/src/runtime/kernel/arm/fp16/reduce_fp16.cc View File

@@ -235,6 +235,6 @@ kernel::LiteKernel *CpuMeanFp16KernelCreator(const std::vector<lite::tensor::Ten
return kernel;
}

REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Reduce, CpuReduceFp16KernelCreator)
REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Mean, CpuMeanFp16KernelCreator)
// REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Reduce, CpuReduceFp16KernelCreator)
// REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Mean, CpuMeanFp16KernelCreator)
} // namespace mindspore::kernel

+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp16/transpose_fp16.cc View File

@@ -206,5 +206,5 @@ kernel::LiteKernel *CpuTransposeFp16KernelCreator(const std::vector<lite::tensor
return kernel;
}

REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Transpose, CpuTransposeFp16KernelCreator)
// REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Transpose, CpuTransposeFp16KernelCreator)
} // namespace mindspore::kernel

Loading…
Cancel
Save