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

Function MakeRuyMatrix

tensorflow/lite/kernels/cpu_backend_gemm_ruy.h:28–43  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

26
27template <typename Scalar, typename DataPointer>
28void MakeRuyMatrix(const MatrixParams<Scalar>& params, DataPointer data_ptr,
29 ruy::Matrix<Scalar>* dst) {
30 dst->layout.rows = params.rows;
31 dst->layout.cols = params.cols;
32 if (params.order == Order::kColMajor) {
33 dst->layout.order = ruy::Order::kColMajor;
34 dst->layout.stride = params.rows;
35 } else {
36 dst->layout.order = ruy::Order::kRowMajor;
37 dst->layout.stride = params.cols;
38 }
39 // Note that ruy::Matrix::data is a ConstCheckingPtr, not a plain pointer.
40 // It does care whether we assign to it a Scalar* or a const Scalar*.
41 dst->data = data_ptr;
42 dst->zero_point = params.zero_point;
43}
44
45template <typename GemmParamsType, typename RuySpecType>
46void MakeRuySpec(const GemmParamsType& params, RuySpecType* ruy_spec) {

Callers 2

ReferenceGemmFunction · 0.85
RunMethod · 0.85

Calls

no outgoing calls

Tested by 1

ReferenceGemmFunction · 0.68