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

Function validate_mm

src/runtime/NEON/functions/NEQLSTMLayer.cpp:46–60  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

44namespace
45{
46Status validate_mm(GEMMLowpOutputStageInfo &gemmlowp_info,
47 const ITensorInfo *mm_input,
48 const ITensorInfo *mm_weights,
49 const ITensorInfo *bias,
50 float gemmlowp_scale,
51 const TensorInfo *mm_res_info,
52 const TensorInfo *outstage_tensor_info)
53{
54 ARM_COMPUTE_RETURN_ON_ERROR(NEGEMMLowpMatrixMultiplyCore::validate(mm_input, mm_weights, nullptr, mm_res_info));
55 ARM_COMPUTE_RETURN_ON_ERROR(quantization::calculate_quantized_multiplier(
56 gemmlowp_scale, &gemmlowp_info.gemmlowp_multiplier, &gemmlowp_info.gemmlowp_shift));
57 ARM_COMPUTE_RETURN_ON_ERROR(
58 NEGEMMLowpOutputStage::validate(mm_res_info, bias, outstage_tensor_info, gemmlowp_info));
59 return Status{};
60}
61} // namespace
62
63Status NEQLSTMLayer::validate_layer_norm(const ITensorInfo &in, const ITensorInfo &weight, const ITensorInfo &bias)

Callers 1

validateMethod · 0.70

Calls 2

validateFunction · 0.50

Tested by

no test coverage detected