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

Method configure_common

src/cpu/kernels/CpuElementwiseKernel.cpp:255–276  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

253}
254
255void CpuArithmeticKernel::configure_common(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
256{
257 ARM_COMPUTE_ERROR_ON_NULLPTR(src0, src1, dst);
258
259 const auto *uk = CpuArithmeticKernel::get_implementation(
260 ElementwiseDataTypeISASelectorData{src0->data_type(), CPUInfo::get().get_isa(), static_cast<int>(_op)});
261
262 ARM_COMPUTE_ERROR_ON(uk == nullptr || uk->ukernel == nullptr);
263
264 _run_method = uk->ukernel;
265 _name = std::string("CpuArithmeticKernel").append("/").append(uk->name);
266
267 // If any of shapes is dynamic, expect a configured window and dst at run-time.
268 if (src0->is_dynamic() || src1->is_dynamic())
269 {
270 return;
271 }
272
273 auto shape_and_window = compute_output_shape_and_window(src0->tensor_shape(), src1->tensor_shape());
274 auto_init_if_empty(*dst, shape_and_window.first, 1, src0->data_type());
275 ICpuKernel::configure(shape_and_window.second);
276}
277
278void CpuComparisonKernel::configure_common(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
279{

Callers

nothing calls this directly

Calls 7

auto_init_if_emptyFunction · 0.85
get_isaMethod · 0.80
appendMethod · 0.80
get_implementationFunction · 0.50
data_typeMethod · 0.45
is_dynamicMethod · 0.45

Tested by

no test coverage detected