Searched refs:set_arrays (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| D | gemv_batched.hpp | 48 void set_arrays(const To *A, const int, const int A_batch_stride, const int A_multi_stride, in set_arrays() function in arm_gemm::GemvBatched
53 _subgemm->set_arrays(A, A_batch_stride, 0, A_multi_stride, in set_arrays()
|
| D | quantize_wrapper.hpp | 74 _subgemm->set_arrays(this->_Aptr, this->_lda, this->_A_batch_stride, this->_A_multi_stride, in set_child_arrays()
121 void set_arrays(const To *A, const int lda, const int A_batch_stride, const int A_multi_stride, in set_arrays() function in arm_gemm::QuantizeWrapper
125 …GemmCommon<To, Tr>::set_arrays(A, lda, A_batch_stride, A_multi_stride, B, ldb, B_multi_stride, C, … in set_arrays()
|
| /external/ComputeLibrary/src/cpu/kernels/assembly/ |
| D | gemm_common.hpp | 181 …virtual void set_arrays(const To *A, const int lda, const int A_batch_stride, const int A_multi_st… in set_arrays() function in arm_gemm::GemmCommon
207 set_arrays(static_cast<const To *>(A), lda, A_batch_stride, A_multi_stride, in set_arrays_generic()
|
| /external/ComputeLibrary/src/cpu/operators/internal/ |
| D | CpuGemmAssemblyDispatch.cpp | 572 _gemm_kernel_asm->set_arrays(in0_ptr, lda, batch_stride_a, multi_stride_a, in run()
|