Home
last modified time | relevance | path

Searched refs:CpuActivation (Results 1 – 20 of 20) sorted by relevance

/external/ComputeLibrary/src/cpu/operators/
DCpuActivation.cpp37 void CpuActivation::configure(const ITensorInfo *input, ITensorInfo *output, const ActivationLayerI… in configure()
45 Status CpuActivation::validate(const ITensorInfo *input, const ITensorInfo *output, const Activatio… in validate()
50 void CpuActivation::run(ITensorPack &tensors) in run()
63 …if(is_validate && !bool(CpuActivation::validate(&src_info.set_is_resizable(false), &dst_info.set_i… in create_activation()
68 auto act_op = std::make_unique<cpu::CpuActivation>(); in create_activation()
DCpuGemm.h145 std::unique_ptr<CpuActivation> _alpha_scale_func{ nullptr };
147 std::unique_ptr<CpuActivation> _activation_func{ nullptr };
DCpuDirectConv3d.cpp64 _activationlayer_function = std::make_unique<CpuActivation>(); in configure()
78 ARM_COMPUTE_RETURN_ON_ERROR(CpuActivation::validate(dst, nullptr, conv_info.act_info)); in validate()
DCpuGemmLowpMatrixMultiplyCore.h49 class CpuActivation; variable
147 std::unique_ptr<CpuActivation> _activation_func;
DCpuDepthwiseConv2d.h139 std::unique_ptr<CpuActivation> _activationlayer_function{ nullptr };
198 … std::unique_ptr<CpuActivation> _activationlayer_function{ nullptr };
DCpuDirectConv2d.cpp81 _activationlayer_function = std::make_unique<CpuActivation>(); in configure()
111 ARM_COMPUTE_RETURN_ON_ERROR(CpuActivation::validate(dst, nullptr, act_info)); in validate()
DCpuActivation.h34 class CpuActivation : public ICpuOperator
DCpuDirectConv3d.h94 std::unique_ptr<CpuActivation> _activationlayer_function;
DCpuDirectConv2d.h96 std::unique_ptr<CpuActivation> _activationlayer_function;
DCpuGemmDirectConv2d.h96 std::unique_ptr<CpuActivation> _activation_func;
DCpuGemm.cpp93 _alpha_scale_func = std::make_unique<cpu::CpuActivation>(); in configure()
148 _activation_func = std::make_unique<cpu::CpuActivation>(); in configure()
277 ARM_COMPUTE_RETURN_ON_ERROR(cpu::CpuActivation::validate(d, nullptr, activation)); in validate()
DCpuDepthwiseConv2d.cpp69 ARM_COMPUTE_RETURN_ON_ERROR(CpuActivation::validate(dst, nullptr, info.act_info)); in validate_arguments_optimized()
133 _activationlayer_function = std::make_unique<cpu::CpuActivation>(); in configure()
329 _activationlayer_function = std::make_unique<cpu::CpuActivation>(); in configure()
365 ARM_COMPUTE_RETURN_ON_ERROR(CpuActivation::validate(dst, nullptr, info.act_info)); in validate()
DCpuWinogradConv2d.h112 std::unique_ptr<CpuActivation> _activation_func;
DCpuGemmDirectConv2d.cpp99 _activation_func(std::make_unique<CpuActivation>()), in CpuGemmDirectConv2d()
DCpuGemmLowpMatrixMultiplyCore.cpp284 _activation_func = std::make_unique<CpuActivation>(); in configure()
499 ARM_COMPUTE_RETURN_ON_ERROR(CpuActivation::validate(output, nullptr, activation)); in validate()
DCpuWinogradConv2d.cpp137 _activation_func(std::make_unique<CpuActivation>()), in CpuWinogradConv2d()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEActivationLayer.cpp36 std::unique_ptr<cpu::CpuActivation> op{ nullptr };
55 _impl->op = std::make_unique<cpu::CpuActivation>(); in configure()
61 return cpu::CpuActivation::validate(input, output, act_info); in validate()
/external/ComputeLibrary/src/
DCMakeLists.txt808 cpu/operators/CpuActivation.cpp
DBUILD.bazel816 "cpu/operators/CpuActivation.cpp",
/external/ComputeLibrary/
DAndroid.bp463 "src/cpu/operators/CpuActivation.cpp",