Searched defs:k_size (Results 1 – 4 of 4) 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()
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 749 const size_t k_size = context->input_size[3]; in xnn_compute_pad_5d() local
|