| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorContractionGpu.h | 24 const Index m_size, const Index n_size, const Index k_size) { in EigenContractionKernelInternal()
513 const Index m_size, const Index n_size, const Index k_size) { in EigenContractionKernel()
538 const Index n_size, const Index k_size, in EigenFloatContractionKernelInternal16x16()
778 const Index n_size, const Index k_size, in EigenFloatContractionKernelInternal()
1145 const Index m_size, const Index n_size, const Index k_size) { in EigenFloatContractionKernel()
1192 const Index m_size, const Index n_size, const Index k_size) { in EigenFloatContractionKernel16x16()
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_contract_sycl.cpp | 37 IndexType k_size, IndexType n_size) { in test_sycl_contraction()
155 IndexType k_size, IndexType n_size) { in test_no_out_of_bounds()
257 void test_scalar(const Device &sycl_device, IndexType m_size, IndexType k_size, in test_scalar()
321 IndexType k_size, IndexType n_size, IndexType m_batch, in contraction_batch()
395 IndexType k_size, IndexType n_size) { in contraction_rhs_transposed()
464 IndexType k_size, IndexType n_size) { in contraction_lhs_transposed()
530 IndexType k_size, IndexType n_size) { in contraction_both_transposed()
|
| H A D | cxx11_tensor_contract_gpu.cu | 26 void test_gpu_contraction(int m_size, int k_size, int n_size) in test_gpu_contraction()
90 void test_scalar(int m_size, int k_size, int n_size) in test_scalar()
|
| H A D | cxx11_tensor_scan_gpu.cu | 25 void test_gpu_cumsum(int m_size, int k_size, int n_size) in test_gpu_cumsum()
|
| H A D | cxx11_tensor_scan_sycl.cpp | 27 IndexType k_size, IndexType n_size, int consume_dim, in test_sycl_cumsum()
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/numeric/ublas/include/boost/numeric/ublas/ |
| D | operation_blocked.hpp | 155 size_type k_size = BOOST_UBLAS_SAME (e1 ().size2 (), e2 ().size1 ()); in block_prod() local
216 size_type k_size = BOOST_UBLAS_SAME (e1 ().size2 (), e2 ().size1 ()); in block_prod() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| H A D | gemm_interleaved_pretransposed_2d.hpp | 466 unsigned int k_size = (current.kmax() - current.k0()); in get_B_pretransposed_array_size() local
490 unsigned int k_size = (current.kmax() - current.k0()); in pretranspose_B_array() local
|
| H A D | gemm_hybrid.hpp | 232 const unsigned int k_size = roundup(kmax-k0, strategy::k_unroll()); in pretranspose_B_array() local
|
| H A D | gemm_hybrid_quantized_inline.hpp | 241 const unsigned int k_size = roundup(kmax-k0, strategy::k_unroll()); in pretranspose_B_array() local
|
| H A D | gemm_hybrid_quantized.hpp | 291 const unsigned int k_size = roundup(kmax-k0, strategy::k_unroll()); in pretranspose_B_array() local
|
| H A D | gemm_hybrid_indirect.hpp | 645 unsigned int k_size = kmax - k0; in pretranspose_B_array() local
|
| H A D | gemm_interleaved.hpp | 1062 unsigned int k_size = (current.kmax() - current.k0()); in pretranspose_B_array() local
|
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A D | sdpa.py | 139 def _try_broadcast_param_size(q_size, k_size, v_size, param_name, debug=False) -> bool: argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/ |
| H A D | sdp_utils_cpp.h | 120 const c10::SymInt k_size, in try_broadcast_param_size()
|
| /aosp_15_r20/external/XNNPACK/src/ |
| H A D | operator-run.c | 965 const size_t k_size = context->input_size[3]; in xnn_compute_pad_5d() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/ash/src/vk/ |
| D | definitions.rs | 40675 pub k_size: u32, field
40738 pub fn k_size(mut self, k_size: u32) -> Self { in k_size() method
|