Searched refs:CpuDirectConv2d (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuDirectConv2d.cpp | 36 CpuDirectConv2d::~CpuDirectConv2d() = default;
38 CpuDirectConv2d::CpuDirectConv2d(std::shared_ptr<IMemoryManager> memory_manager) in CpuDirectConv2d() function in arm_compute::cpu::CpuDirectConv2d
44 void CpuDirectConv2d::configure(ITensorInfo *src, ITensorInfo *weights, const ITensorInfo *bias, IT… in configure()
86 Status CpuDirectConv2d::validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorI… in validate()
117 void CpuDirectConv2d::run(ITensorPack &tensors) in run()
|
| D | CpuDirectConv2d.h | 55 class CpuDirectConv2d : public ICpuOperator
58 CpuDirectConv2d(std::shared_ptr<IMemoryManager> memory_manager = nullptr);
59 ~CpuDirectConv2d();
|
| D | CpuConv2d.cpp | 82 auto f = std::make_unique<CpuDirectConv2d>(); in configure()
113 …ARM_COMPUTE_RETURN_ON_ERROR(CpuDirectConv2d::validate(input, weights, biases, output, conv_info, a… in validate()
177 && (CpuDirectConv2d::validate(input, weights, nullptr, output, conv_info, act_info))) in get_convolution_method()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEDirectConvolutionLayer.cpp | 40 std::unique_ptr<cpu::CpuDirectConv2d> op{ nullptr };
55 _impl->op = std::make_unique<cpu::CpuDirectConv2d>(_memory_manager); in configure()
62 return cpu::CpuDirectConv2d::validate(input, weights, bias, output, conv_info, act_info); in validate()
|
| /external/ComputeLibrary/src/ |
| D | CMakeLists.txt | 819 cpu/operators/CpuDirectConv2d.cpp
|
| D | BUILD.bazel | 827 "cpu/operators/CpuDirectConv2d.cpp",
|
| /external/ComputeLibrary/ |
| D | Android.bp | 474 "src/cpu/operators/CpuDirectConv2d.cpp",
|