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

Method has_opt_impl

src/cpu/operators/CpuFullyConnected.cpp:362–377  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

360}
361
362Status CpuFullyConnected::has_opt_impl(arm_compute::WeightFormat &expected_weight_format,
363 const ITensorInfo *src,
364 const ITensorInfo *weights,
365 const ITensorInfo *biases,
366 const ITensorInfo *dst,
367 FullyConnectedLayerInfo fc_info,
368 WeightsInfo weights_info)
369{
370 GEMMInfo gemm_info;
371 gemm_info.set_activation_info(fc_info.activation_info);
372 gemm_info.set_fast_math(fc_info.enable_fast_math);
373 gemm_info.set_fixed_format(weights_info.weight_format() != WeightFormat::UNSPECIFIED);
374 gemm_info.set_weight_format(weights_info.weight_format());
375
376 return CpuGemm::has_opt_impl(expected_weight_format, src, weights, biases, dst, gemm_info);
377}
378
379Status CpuFullyConnected::validate(const ITensorInfo *src,
380 const ITensorInfo *weights,

Callers

nothing calls this directly

Calls 5

set_activation_infoMethod · 0.80
set_fast_mathMethod · 0.80
set_fixed_formatMethod · 0.80
weight_formatMethod · 0.45
set_weight_formatMethod · 0.45

Tested by

no test coverage detected