forked from mindspore-Ecosystem/mindspore
rename arm to cpu
This commit is contained in:
parent
7c88495a74
commit
c349c1f2d0
|
|
@ -32,8 +32,8 @@ mindspore/mindspore/lite/nnacl/fp16/pack_fp16.c:PackNHWCToNCHWFp16
|
|||
mindspore/mindspore/lite/src/ios_reg_ops.cc:mindspore::lite::IosRegisterOps
|
||||
mindspore/mindspore/lite/src/ios_reg_kernels.h:mindspore::kernel::IosRegisterKernels
|
||||
mindspore/mindspore/lite/nnacl/infer/infer_register.c:RegisterInfer
|
||||
mindspore/mindspore/lite/src/runtime/kernel/arm/base/quant_dtype_cast.cc:mindspore::kernel::QuantDTypeCastCPUKernel::QuantDTypeCast
|
||||
mindspore/mindspore/lite/src/runtime/kernel/arm/base/quant_dtype_cast.cc:mindspore::kernel::QuantDTypeCastCPUKernel::Run
|
||||
mindspore/mindspore/lite/src/runtime/kernel/cpu/base/quant_dtype_cast.cc:mindspore::kernel::QuantDTypeCastCPUKernel::QuantDTypeCast
|
||||
mindspore/mindspore/lite/src/runtime/kernel/cpu/base/quant_dtype_cast.cc:mindspore::kernel::QuantDTypeCastCPUKernel::Run
|
||||
mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/prim.cc:mindspore::abstract::ConvertAbstractToPython
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/strided_slice_infer.c:StridedSliceInferShape
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/winograd_transform_fp16.c:WinogradInputTransformFp16
|
||||
|
|
@ -175,7 +175,7 @@ mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-genera
|
|||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_12x8_kernel_nc8hw8_fp32.c:nnacl_gemm_fma_12x8_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_5x8_kernel_nc8hw8_fp32.c:nnacl_gemm_fma_5x8_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_3x8_kernel_nc8hw8_fp32_asm.c:nnacl_gemm_fma_3x8_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/matmul_fp32_base.cc:mindspore::kernel::MatmulFp32BaseCPUKernel::init_global_variable
|
||||
mindspore/mindspore/lite/src/runtime/kernel/cpu/fp32/matmul_fp32_base.cc:mindspore::kernel::MatmulFp32BaseCPUKernel::init_global_variable
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_3x24_kernel_nc8hw8_fp32_asm.c:nnacl_gemm_fma_3x24_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_4x16_kernel_nc8hw8_fp32_asm.c:nnacl_gemm_fma_4x16_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_1x24_kernel_nc8hw8_fp32_asm.c:nnacl_gemm_fma_1x24_kernel_nc8hw8_fp32
|
||||
|
|
@ -191,11 +191,12 @@ mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-genera
|
|||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_2x32_kernel_nc8hw8_fp32.c:nnacl_gemm_fma_2x32_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_2x24_kernel_nc8hw8_fp32.c:nnacl_gemm_fma_2x24_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/HPC-generator/gemm_fma/nnacl_gemm_fma_1x32_kernel_nc8hw8_fp32_asm.c:nnacl_gemm_fma_1x32_kernel_nc8hw8_fp32
|
||||
mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/matmul_fp32_base.cc:mindspore::kernel::MatmulFp32BaseCPUKernel::Run
|
||||
mindspore/mindspore/lite/src/runtime/kernel/cpu/fp32/matmul_fp32_base.cc:mindspore::kernel::MatmulFp32BaseCPUKernel::Run
|
||||
mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_partition.cc:mindspore::parallel::GetWeights
|
||||
mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_partition.cc:mindspore::parallel::PartitionNode
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/instance_norm_fp16.c:InstanceNormNC8HW8Fp16
|
||||
mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/matmul_fp32_base.cc:mindspore::kernel::MatmulFp32BaseCPUKernel::init_global_variable
|
||||
mindspore/mindspore/lite/src/runtime/kernel/cpu/fp32/matmul_fp32_base.cc:mindspore::kernel::MatmulFp32BaseCPUKernel::init_global_variable
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/conv_winograd_fp32.c:ConvWinogardFp32
|
||||
mindspore/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/lamb_next_mv_with_decay_v1_rule.cc:mindspore::opt::MatchAdd5Pattern
|
||||
mindspore/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/experimental/conv_fp32_nchwx_avx512.c:conv2d_compute_fp32_nchwx_avx512
|
||||
mindspore/mindspore/lite/src/runtime/kernel/cpu/control/tensorlist_setitem.cc:mindspore::kernel::TensorListSetItemCPUKernel::Run
|
||||
|
|
|
|||
|
|
@ -23,8 +23,8 @@ MS_ALL:
|
|||
mindspore/ccsrc/CMakeLists.txt
|
||||
mindspore/lite/java
|
||||
mindspore/lite/src/train
|
||||
mindspore/lite/src/runtime/kernel/arm/fp32
|
||||
mindspore/lite/src/runtime/kernel/arm/fp32_grad
|
||||
mindspore/lite/src/runtime/kernel/cpu/fp32
|
||||
mindspore/lite/src/runtime/kernel/cpu/fp32_grad
|
||||
mindspore/ccsrc/backend/kernel_compiler
|
||||
mindspore/ccsrc/fl
|
||||
mindspore/ccsrc/ps
|
||||
|
|
|
|||
|
|
@ -512,7 +512,7 @@ include_directories(${CORE_DIR})
|
|||
include_directories(${CORE_DIR}/ir)
|
||||
include_directories(${CCSRC_DIR})
|
||||
include_directories(${CMAKE_CURRENT_SOURCE_DIR})
|
||||
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/runtime/kernel/arm)
|
||||
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/runtime/kernel/cpu)
|
||||
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/)
|
||||
include_directories(${TOP_DIR}/third_party)
|
||||
include_directories(${CMAKE_BINARY_DIR})
|
||||
|
|
|
|||
|
|
@ -342,7 +342,7 @@ if(MSLITE_ENABLE_GRAPH_KERNEL)
|
|||
set(LITE_SRC ${LITE_SRC} ${GRAPH_KERNEL_SRC})
|
||||
endif()
|
||||
|
||||
add_subdirectory(runtime/kernel/arm)
|
||||
add_subdirectory(runtime/kernel/cpu)
|
||||
|
||||
add_library(lite_src_mid OBJECT ${LITE_SRC})
|
||||
add_dependencies(lite_src_mid fbs_src)
|
||||
|
|
|
|||
|
|
@ -21,7 +21,7 @@
|
|||
#include "include/errorcode.h"
|
||||
#include "src/common/log_adapter.h"
|
||||
#ifdef ENABLE_FP16
|
||||
#include "src/runtime/kernel/arm/fp16/fp16_op_handler.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/fp16_op_handler.h"
|
||||
#endif
|
||||
|
||||
namespace mindspore {
|
||||
|
|
|
|||
|
|
@ -22,7 +22,7 @@
|
|||
#include "src/common/tensor_util.h"
|
||||
#include "src/runtime/inner_allocator.h"
|
||||
#ifdef ENABLE_FP16
|
||||
#include "src/runtime/kernel/arm/fp16/fp16_op_handler.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/fp16_op_handler.h"
|
||||
#endif
|
||||
namespace {
|
||||
const constexpr int kSwitchMaxInputKernelSize = 3;
|
||||
|
|
|
|||
|
|
@ -18,7 +18,7 @@
|
|||
#include <algorithm>
|
||||
#include <set>
|
||||
#include "src/kernel_exec_util.h"
|
||||
#include "src/runtime/kernel/arm/base/partial_fusion.h"
|
||||
#include "src/runtime/kernel/cpu/base/partial_fusion.h"
|
||||
#include "nnacl/call_parameter.h"
|
||||
#include "src/control_flow/exit_subgraph_kernel.h"
|
||||
#include "src/control_flow/identity_kernel.h"
|
||||
|
|
|
|||
|
|
@ -28,7 +28,7 @@
|
|||
#ifndef CONTROLFLOW_TENSORLIST_CLIP
|
||||
#include "src/control_flow/entrance_subgraph_kernel.h"
|
||||
#include "src/control_flow/exit_subgraph_kernel.h"
|
||||
#include "src/runtime/kernel/arm/base/partial_fusion.h"
|
||||
#include "src/runtime/kernel/cpu/base/partial_fusion.h"
|
||||
#endif
|
||||
|
||||
namespace mindspore::kernel {
|
||||
|
|
|
|||
|
|
@ -22,7 +22,7 @@
|
|||
#include "src/common/tensor_util.h"
|
||||
#include "src/common/common.h"
|
||||
#include "src/runtime/inner_allocator.h"
|
||||
#include "src/runtime/kernel/arm/base/partial_fusion.h"
|
||||
#include "src/runtime/kernel/cpu/base/partial_fusion.h"
|
||||
#ifndef CONTROLFLOW_TENSORLIST_CLIP
|
||||
#include "src/control_flow/actor/switch_actor.h"
|
||||
#include "src/control_flow/actor/entrance_actor.h"
|
||||
|
|
|
|||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/argminmax_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/argminmax_base.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/arithmetic_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/arithmetic_base.h"
|
||||
#include <map>
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/assert.h"
|
||||
#include "src/runtime/kernel/cpu/base/assert.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/call.h"
|
||||
#include "src/runtime/kernel/cpu/base/call.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
#ifndef CONTROLFLOW_TENSORLIST_CLIP
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/concat_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/concat_base.h"
|
||||
#include <algorithm>
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/constant_of_shape.h"
|
||||
#include "src/runtime/kernel/cpu/base/constant_of_shape.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/convolution_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/convolution_base.h"
|
||||
#include <cfloat>
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -32,7 +32,7 @@
|
|||
#include "src/pack_weight_manager.h"
|
||||
#endif
|
||||
#include "include/context.h"
|
||||
#include "src/runtime/kernel/arm/base/layout_transform.h"
|
||||
#include "src/runtime/kernel/cpu/base/layout_transform.h"
|
||||
#include "src/weight_decoder.h"
|
||||
#include "include/errorcode.h"
|
||||
|
||||
|
|
@ -13,9 +13,9 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/crop_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/crop_base.h"
|
||||
#include <vector>
|
||||
#include "src/runtime/kernel/arm/fp32/crop_fp32.h"
|
||||
#include "src/runtime/kernel/cpu/fp32/crop_fp32.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/detection_post_process_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/detection_post_process_base.h"
|
||||
#include <cfloat>
|
||||
#include <cmath>
|
||||
#include <vector>
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/gather_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/gather_base.h"
|
||||
#include <algorithm>
|
||||
|
||||
using mindspore::lite::RET_ERROR;
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/group_convolution_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/group_convolution_base.h"
|
||||
#include "src/runtime/infer_manager.h"
|
||||
#include "include/errorcode.h"
|
||||
|
||||
|
|
@ -21,9 +21,9 @@
|
|||
#include <vector>
|
||||
#include "src/lite_kernel.h"
|
||||
#include "nnacl/op_base.h"
|
||||
#include "src/runtime/kernel/arm/base/convolution_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/convolution_base.h"
|
||||
#include "nnacl/fp32/conv_common_fp32.h"
|
||||
#include "src/runtime/kernel/arm/base/group_convolution_creator.h"
|
||||
#include "src/runtime/kernel/cpu/base/group_convolution_creator.h"
|
||||
|
||||
namespace mindspore::kernel {
|
||||
class GroupConvolutionBaseCPUKernel : public ConvolutionBaseCPUKernel {
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/group_convolution_creator.h"
|
||||
#include "src/runtime/kernel/cpu/base/group_convolution_creator.h"
|
||||
|
||||
namespace mindspore::kernel {
|
||||
void CopyTensorQuantParam(lite::Tensor *dst, const lite::Tensor *src) {
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/layout_transform.h"
|
||||
#include "src/runtime/kernel/cpu/base/layout_transform.h"
|
||||
#include "src/common/log_adapter.h"
|
||||
#include "schema/ops_types_generated.h"
|
||||
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/one_hot_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/one_hot_base.h"
|
||||
#include "nnacl/fp32/one_hot_fp32.h"
|
||||
#if defined(ENABLE_ARM) && defined(ENABLE_FP16)
|
||||
#include "nnacl/fp16/one_hot_fp16.h"
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/partial_fusion.h"
|
||||
#include "src/runtime/kernel/cpu/base/partial_fusion.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
#ifndef CONTROLFLOW_TENSORLIST_CLIP
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/pooling_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/pooling_base.h"
|
||||
#include <cfloat>
|
||||
#include <cmath>
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -16,7 +16,7 @@
|
|||
|
||||
#include <vector>
|
||||
#include <cmath>
|
||||
#include "src/runtime/kernel/arm/base/prior_box.h"
|
||||
#include "src/runtime/kernel/cpu/base/prior_box.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/quant_dtype_cast.h"
|
||||
#include "src/runtime/kernel/cpu/base/quant_dtype_cast.h"
|
||||
#include <vector>
|
||||
#include "nnacl/int8/quant_dtype_cast_int8.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/random_normal.h"
|
||||
#include "src/runtime/kernel/cpu/base/random_normal.h"
|
||||
#include <random>
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
|
@ -14,11 +14,11 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/reduce_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/reduce_base.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "src/runtime/kernel/arm/fp32/reduce_fp32.h"
|
||||
#include "src/runtime/kernel/cpu/fp32/reduce_fp32.h"
|
||||
|
||||
using mindspore::kernel::KERNEL_ARCH;
|
||||
using mindspore::lite::KernelRegistrar;
|
||||
|
|
@ -15,7 +15,7 @@
|
|||
*/
|
||||
|
||||
#include <cstring>
|
||||
#include "src/runtime/kernel/arm/base/reshape_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/reshape_base.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
|
@ -15,7 +15,7 @@
|
|||
*/
|
||||
|
||||
#include <vector>
|
||||
#include "src/runtime/kernel/arm/base/resize_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/resize_base.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/scatter_nd_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/scatter_nd_base.h"
|
||||
#include <cstring>
|
||||
#include <vector>
|
||||
#include "schema/model_generated.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/select.h"
|
||||
#include "src/runtime/kernel/cpu/base/select.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
#ifndef CONTROLFLOW_TENSORLIST_CLIP
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/slice_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/slice_base.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "nnacl/base/slice_base.h"
|
||||
#include "src/tensor.h"
|
||||
|
|
@ -14,9 +14,9 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/softmax_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/softmax_base.h"
|
||||
#include <vector>
|
||||
#include "src/runtime/kernel/arm/fp32/softmax_fp32.h"
|
||||
#include "src/runtime/kernel/cpu/fp32/softmax_fp32.h"
|
||||
#include "nnacl/fp32/softmax_fp32.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/split_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/split_base.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/split_with_over_lap_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/split_with_over_lap_base.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "src/tensor.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/stack_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/stack_base.h"
|
||||
#include <vector>
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/strided_slice.h"
|
||||
#include "src/runtime/kernel/cpu/base/strided_slice.h"
|
||||
#include <vector>
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/base/tile_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/tile_base.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "nnacl/nnacl_common.h"
|
||||
#include "include/errorcode.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/base/transpose_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/transpose_base.h"
|
||||
|
||||
using mindspore::lite::RET_ERROR;
|
||||
using mindspore::lite::RET_OK;
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/control/switch.h"
|
||||
#include "src/runtime/kernel/cpu/control/switch.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/control/switch_layer.h"
|
||||
#include "src/runtime/kernel/cpu/control/switch_layer.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
|
||||
|
|
@ -16,7 +16,7 @@
|
|||
|
||||
#include <vector>
|
||||
#include "include/errorcode.h"
|
||||
#include "src/runtime/kernel/arm/control/tensor_array.h"
|
||||
#include "src/runtime/kernel/cpu/control/tensor_array.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "src/tensorlist.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/control/tensorlist_fromtensor.h"
|
||||
#include "src/runtime/kernel/cpu/control/tensorlist_fromtensor.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/control/tensorlist_getitem.h"
|
||||
#include "src/runtime/kernel/cpu/control/tensorlist_getitem.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "include/ms_tensor.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/control/tensorlist_reserve.h"
|
||||
#include "src/runtime/kernel/cpu/control/tensorlist_reserve.h"
|
||||
#include <vector>
|
||||
#include "include/errorcode.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -14,7 +14,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/control/tensorlist_setitem.h"
|
||||
#include "src/runtime/kernel/cpu/control/tensorlist_setitem.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "include/ms_tensor.h"
|
||||
#include "src/kernel_registry.h"
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/control/tensorlist_stack.h"
|
||||
#include "src/runtime/kernel/cpu/control/tensorlist_stack.h"
|
||||
#include <functional>
|
||||
#include <vector>
|
||||
#include "include/errorcode.h"
|
||||
|
|
@ -14,11 +14,11 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/fp16/activation_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/activation_fp16.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "src/runtime/kernel/arm/fp16/common_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/common_fp16.h"
|
||||
|
||||
using mindspore::kernel::KERNEL_ARCH;
|
||||
using mindspore::lite::KernelRegistrar;
|
||||
|
|
@ -13,7 +13,7 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/fp16/addn_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/addn_fp16.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "nnacl/fp16/arithmetic_fp16.h"
|
||||
|
||||
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/fp16/arithmetic_compare_fp16.h"
|
||||
#include "src/runtime/kernel/arm/fp16/common_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/arithmetic_compare_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/common_fp16.h"
|
||||
#include "nnacl/fp16/arithmetic_fp16.h"
|
||||
#include "nnacl/fp16/cast_fp16.h"
|
||||
#include "schema/model_generated.h"
|
||||
|
|
@ -14,9 +14,9 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/runtime/kernel/arm/fp16/arithmetic_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/arithmetic_fp16.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "src/runtime/kernel/arm/fp16/common_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/common_fp16.h"
|
||||
#include "nnacl/fp16/arithmetic_fp16.h"
|
||||
#include "nnacl/fp16/cast_fp16.h"
|
||||
|
||||
|
|
@ -18,7 +18,7 @@
|
|||
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_FP16_ARITHMETIC_FP16_H_
|
||||
|
||||
#include <vector>
|
||||
#include "src/runtime/kernel/arm/base/arithmetic_base.h"
|
||||
#include "src/runtime/kernel/cpu/base/arithmetic_base.h"
|
||||
|
||||
namespace mindspore::kernel {
|
||||
class ArithmeticFP16CPUKernel : public ArithmeticBaseCPUKernel {
|
||||
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "src/runtime/kernel/arm/fp16/arithmetic_self_fp16.h"
|
||||
#include "src/runtime/kernel/arm/fp16/common_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/arithmetic_self_fp16.h"
|
||||
#include "src/runtime/kernel/cpu/fp16/common_fp16.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "nnacl/fp16/arithmetic_self_fp16.h"
|
||||
|
||||
|
|
@ -17,7 +17,7 @@
|
|||
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_FP16_ARITHMETIC_SELF_FP16_H_
|
||||
|
||||
#include <vector>
|
||||
#include "src/runtime/kernel/arm/fp32/arithmetic_self_fp32.h"
|
||||
#include "src/runtime/kernel/cpu/fp32/arithmetic_self_fp32.h"
|
||||
|
||||
namespace mindspore::kernel {
|
||||
typedef int (*ArithmeticSelfFp16Func)(const float16_t *input, float16_t *output, int element_size);
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue