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

Method configure

src/cpu/kernels/CpuDynamicGemmKernel.cpp:45–66  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

43{
44
45void CpuDynamicGemmKernel::configure(const ITensorInfo *a,
46 const ITensorInfo *b,
47 const ITensorInfo *c,
48 ITensorInfo *d,
49 float alpha,
50 float beta,
51 size_t base_aux_slot,
52 const GEMMInfo &gemm_info)
53{
54 ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "CpuDynamicGemmKernel::configure");
55 ARM_COMPUTE_ERROR_THROW_ON(CpuDynamicGemmKernel::validate(a, b, c, d, alpha, beta, gemm_info));
56
57 _heuristics = CpuDynamicGemmKernelHeuristics{a, b, c, d, alpha, beta, gemm_info};
58
59 _name = std::string{"CpuDynamicGemmKernel"}.append("/").append(_heuristics.name());
60
61 _base_aux_slot = base_aux_slot;
62 _aux_mem.reserve(Count);
63
64 Window window = _heuristics.get_window()(d);
65 ICPPKernel::configure(window);
66}
67
68Status CpuDynamicGemmKernel::validate(const ITensorInfo *a,
69 const ITensorInfo *b,

Callers

nothing calls this directly

Calls 4

appendMethod · 0.80
get_windowMethod · 0.80
validateFunction · 0.50
nameMethod · 0.45

Tested by

no test coverage detected