Searched defs:k_size (Results 1 – 3 of 3) sorted by relevance
/external/eigen/unsupported/test/ |
D | cxx11_tensor_contract_cuda.cu | 27 void test_cuda_contraction(int m_size, int k_size, int n_size) in test_cuda_contraction() 91 void test_scalar(int m_size, int k_size, int n_size) in test_scalar()
|
D | cxx11_tensor_scan_cuda.cu | 26 void test_cuda_cumsum(int m_size, int k_size, int n_size) in test_cuda_cumsum()
|
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | TensorContractionCuda.h | 24 const Index m_size, const Index n_size, const Index k_size) { in EigenContractionKernelInternal() 505 const Index m_size, const Index n_size, const Index k_size) { in EigenContractionKernel() 530 const Index n_size, const Index k_size, in EigenFloatContractionKernelInternal16x16() 767 const Index n_size, const Index k_size, in EigenFloatContractionKernelInternal() 1133 const Index m_size, const Index n_size, const Index k_size) { in EigenFloatContractionKernel() 1179 const Index m_size, const Index n_size, const Index k_size) { in EigenFloatContractionKernel16x16()
|