MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MakeRuySpec

Function MakeRuySpec

tensorflow/lite/kernels/cpu_backend_gemm_ruy.h:46–61  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

44
45template <typename GemmParamsType, typename RuySpecType>
46void MakeRuySpec(const GemmParamsType& params, RuySpecType* ruy_spec) {
47 // This validation has already been performed by the Gemm API entry point,
48 // but it doesn't hurt to test specifically this again here, where it's
49 // being used.
50 ValidateGemmParams(params);
51
52 ruy_spec->multiplier_fixedpoint = params.multiplier_fixedpoint;
53 ruy_spec->multiplier_exponent = params.multiplier_exponent;
54 ruy_spec->multiplier_fixedpoint_perchannel =
55 params.multiplier_fixedpoint_perchannel;
56 ruy_spec->multiplier_exponent_perchannel =
57 params.multiplier_exponent_perchannel;
58 ruy_spec->bias = params.bias;
59 ruy_spec->clamp_min = params.clamp_min;
60 ruy_spec->clamp_max = params.clamp_max;
61}
62
63template <typename LhsScalar, typename RhsScalar, typename AccumScalar,
64 typename DstScalar, QuantizationFlavor quantization_flavor>

Callers 2

ReferenceGemmFunction · 0.85
RunMethod · 0.85

Calls 1

ValidateGemmParamsFunction · 0.85

Tested by 1

ReferenceGemmFunction · 0.68