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

Method Run

tensorflow/lite/kernels/cpu_backend_gemm_ruy.h:66–84  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

64 typename DstScalar, QuantizationFlavor quantization_flavor>
65struct GemmImplUsingRuy {
66 static void Run(
67 const MatrixParams<LhsScalar>& lhs_params, const LhsScalar* lhs_data,
68 const MatrixParams<RhsScalar>& rhs_params, const RhsScalar* rhs_data,
69 const MatrixParams<DstScalar>& dst_params, DstScalar* dst_data,
70 const GemmParams<AccumScalar, DstScalar, quantization_flavor>& params,
71 CpuBackendContext* context) {
72 ruy::Matrix<LhsScalar> ruy_lhs;
73 ruy::Matrix<RhsScalar> ruy_rhs;
74 ruy::Matrix<DstScalar> ruy_dst;
75 MakeRuyMatrix(lhs_params, lhs_data, &ruy_lhs);
76 MakeRuyMatrix(rhs_params, rhs_data, &ruy_rhs);
77 MakeRuyMatrix(dst_params, dst_data, &ruy_dst);
78
79 ruy::BasicSpec<AccumScalar, DstScalar> ruy_spec;
80 MakeRuySpec(params, &ruy_spec);
81
82 ruy::Mul<ruy::kAllPaths>(ruy_lhs, ruy_rhs, ruy_spec, context->ruy_context(),
83 &ruy_dst);
84 }
85};
86
87} // namespace detail

Callers

nothing calls this directly

Calls 3

MakeRuyMatrixFunction · 0.85
MakeRuySpecFunction · 0.85
ruy_contextMethod · 0.80

Tested by

no test coverage detected