Home
last modified time | relevance | path

Searched refs:input_row_start (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/
H A Dmp8x27-sse2.c55 const uint8_t** input_row_start = input; in pytorch_q8dwconv_ukernel_mp8x27__sse2() local
157 input = (const uint8_t**)((uint8_t*)input_row_start + input_row_stride); in pytorch_q8dwconv_ukernel_mp8x27__sse2()
H A Dmp8x27-neon.c56 const uint8_t** input_row_start = input; in pytorch_q8dwconv_ukernel_mp8x27__neon() local
1042 input = (const uint8_t**)((uintptr_t)input_row_start + input_row_stride); in pytorch_q8dwconv_ukernel_mp8x27__neon()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthwise_conv_op_gpu.h121 const int input_row_start = out_row * stride - pad_height;
123 const int input_row_end = input_row_start + filter_height;
129 if (input_row_start >= 0 && input_col_start >= 0 &&
133 const int in_row = input_row_start + filter_row;
153 const int in_row = input_row_start + filter_row;
408 const int input_row_start = out_row * stride - pad_height;
410 const int input_row_end = input_row_start + filter_height;
414 if (input_row_start >= 0 && input_col_start >= 0 &&
419 const int in_row = input_row_start + filter_row;
439 const int in_row = input_row_start + filter_row;
H A Dconv_ops_using_gemm.cc386 const T1* input_row_start = in operator ()() local
390 input_row_start + (center_copy_count * input_depth); in operator ()()
393 std::copy(input_row_start, input_row_end, im2col_center_start); in operator ()()
H A Dquantized_conv_ops.cc364 const T1* input_row_start = in operator ()() local
369 memcpy(im2col_center_start, input_row_start, in operator ()()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dscale_and_translate_op.cc494 const T* input_row_start = image + in_row_size * y; in GatherColumns() local
497 const T* in_pix = input_row_start + starts[x] * channels; in GatherColumns()