MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / configure

Method configure

src/gpu/cl/operators/ClPool3d.cpp:36–52  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

34namespace opencl
35{
36void ClPool3d::configure(const ClCompileContext &compile_context,
37 const ITensorInfo *src,
38 ITensorInfo *dst,
39 const Pooling3dLayerInfo &info)
40{
41 ARM_COMPUTE_ERROR_ON_NULLPTR(src);
42 ARM_COMPUTE_LOG_PARAMS(src, dst, info);
43
44 // Configure pooling kernel
45 auto k = std::make_unique<kernels::ClPool3dKernel>();
46 k->set_target(CLScheduler::get().target());
47 k->configure(compile_context, src, dst, info);
48 _kernel = std::move(k);
49
50 // Tune kernels
51 CLScheduler::get().tune_kernel_static(*_kernel);
52}
53
54Status ClPool3d::validate(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &info)
55{

Callers

nothing calls this directly

Calls 3

set_targetMethod · 0.45
targetMethod · 0.45
tune_kernel_staticMethod · 0.45

Tested by

no test coverage detected