Home
last modified time | relevance | path

Searched defs:col_data (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSpmmReduceKernel.cpp84 auto col_data = col_indices.accessor<const index_t, 1>(); in spmm_reduce_kernel_impl() local
182 auto col_data = col_indices.accessor<const index_t, 1>(); in spmm_reduce_arg_kernel_impl() local
262 auto col_data = col_indices.accessor<const index_t, 1>(); in spmm_reduce_backward_input_kernel_impl() local
311 auto col_data = col_indices.accessor<const index_t, 1>(); in spmm_reduce_backward_input_arg_kernel_impl() local
393 auto col_data = col_indices.accessor<const index_t, 1>(); in spmm_reduce_backward_other_arg_kernel_impl() local
/aosp_15_r20/external/libgav1/src/dsp/x86/
H A Dintrapred_sse4.cc391 const __m128i col_data = Load4(column); in ColStore4_SSE4_1() local
402 const __m128i col_data = LoadLo8(column); in ColStore8_SSE4_1() local
417 const __m128i col_data = _mm_loadu_si128(static_cast<const __m128i*>(column)); in ColStore16_SSE4_1() local
441 const __m128i col_data = in ColStore32_SSE4_1() local
471 const __m128i col_data = in ColStore64_SSE4_1() local
1952 const __m128i col_data = LoadLo8(column); in ColStore4_SSE4_1() local
1961 const __m128i col_data = LoadUnaligned16(column); in ColStore8_SSE4_1() local
1978 const __m128i col_data = in ColStore16_SSE4_1() local
1996 const __m128i col_data = in ColStore32_SSE4_1() local
2014 const __m128i col_data = in ColStore64_SSE4_1() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_input_ops.h78 void Col2im(const T* col_data, const int depth, const int height, in Col2im()
H A Dconv_grad_ops_3d.cc78 void Col2im(const T* col_data, const int depth, const int planes, in Col2im()
132 const int stride_w, T* col_data) { in Im2col()
H A Dconv_grad_filter_ops.cc77 const int stride_h, const int stride_w, T* col_data) { in Im2col()
/aosp_15_r20/external/intel-media-driver/media_driver/linux/common/codec/ddi/
H A Dmedia_libvpx_vp9.cpp446 int col_data; in Vp9WriteUncompressHeader() local
/aosp_15_r20/external/intel-media-driver/media_softlet/linux/common/codec/ddi/enc/
H A Dmedia_libvpx_vp9_next.cpp446 int col_data; in Vp9WriteUncompressHeader() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h4900 void Col2im(const T* col_data, const int depth, const int height, in Col2im()
8147 void Col2im(const T* col_data, const int channel, const int planes, in Col2im()