Home
last modified time | relevance | path

Searched defs:weight_row (Results 1 – 3 of 3) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/interleaves/
D8b_mla.cpp62 auto weight_row = weights + c; in generic_pack() local
/external/pytorch/aten/src/ATen/native/cuda/
DEmbedding.cu161 const int weight_row = ((int) input[idx]) * stride; in embedding_backward_kernel() local
DIndexing.cu83 const int64_t weight_row = ((int64_t) sorted_indices[idx]) * stride + z * stride_before; in indexing_backward_kernel() local
148 const int64_t weight_row = crnt_sorted_idx * stride + z * stride_before; in indexing_backward_kernel_stride_1() local
206 const int64_t weight_row = crnt_sorted_idx * stride + z * stride_before; in indexing_backward_kernel_small_stride() local
246 const int64_t weight_row = ((int64_t) sorted_indices[idx]) * stride + z * stride_before; in indexing_backward_kernel_quantized() local