Home
last modified time | relevance | path

Searched defs:row_ptr (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/
H A Dpooling_depthfirst_generic.hpp172 auto row_ptr = input.base + input_i*input.ld_row + input_j*input.ld_col + channel_start; in compute_tile_padded() local
232 auto row_ptr = input.base + input_i*input.ld_row + input_j*input.ld_col + channel_start; in compute_row_padded_tile_row() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dkernels_gpu.cu.cc371 const int* row_ptr, in CSRSparseMatrixSoftmaxKernel2D()
397 const int* row_ptr, const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel3D()
424 auto row_ptr = logits.row_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGPUImpl() local
H A Dsparse_matrix.h618 TTypes<int32>::UnalignedConstVec row_ptr; member
626 TTypes<int32>::UnalignedVec row_ptr; member
H A Dcsr_sparse_matrix_to_dense_op.cc77 auto row_ptr = csr_sparse_matrix->row_pointers().vec<int32>(); in Compute() local
/aosp_15_r20/external/deqp/external/openglcts/modules/gles31/
H A Des31cTextureStorageMultisampleFunctionalTests.cpp667 float *row_ptr = (float *)buffer + row_width * data_y; in iterate() local
1054 const unsigned int *row_ptr = (unsigned int *)((char *)buffer + data_y * row_width); in iterate() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dsse_tensor_utils.cc211 const int8_t* __restrict__ row_ptr = matrix + row * m_cols; in Avx2MatrixBatchVectorMultiplyAccumulateImpl() local
624 const int8_t* __restrict__ row_ptr = input_vector + row * reduction_size; in SseReductionSumVector() local
H A Dneon_tensor_utils.cc810 int8_t* row_ptr = in NeonMatrixBatchVectorMultiplyImpl() local
1131 int8_t* row_ptr = (int8_t*)matrix + row * m_cols; // NOLINT in NeonMatrixBatchVectorMultiplyAccumulate() local
1253 const int8_t* row_ptr = matrix + i * n_col; in NeonMatrixScalarMultiplyAccumulate() local
1321 int8_t* row_ptr = (int8_t*)matrix + row * m_cols; // NOLINT in NeonMatrixBatchVectorMultiplyAccumulateImpl() local
2102 const int8_t* row_ptr = matrix; in NeonSparseMatrixBatchVectorMultiplyAccumulate() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/input_transforms/
H A Darm_fp32_6x6.cpp51 const float* const row_ptr = input_base + xi*input_row_stride; in arm_fp32_6x6() local
H A Da64_fp16_6x6.cpp50 const __fp16* const row_ptr = input_base + xi*input_row_stride; in a64_fp16_6x6() local
H A Darm_fp32_4x4.cpp48 const float* const row_ptr = input_base + xi*input_row_stride; in arm_fp32_4x4() local
/aosp_15_r20/external/libpng/contrib/oss-fuzz/
H A Dlibpng_read_fuzzer.cc54 png_voidp row_ptr = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dportable_tensor_utils.cc145 const int8_t* row_ptr = matrix; in PortableMatrixBatchVectorMultiplyAccumulate() local
184 const int8_t* row_ptr = matrix; in PortableMatrixBatchVectorMultiplyAccumulate() local
305 const int8_t* row_ptr = matrix; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dconvolver.hpp106 std::tuple<unsigned int, unsigned int> next_block(const T ** const row_ptr) { in next_block()
/aosp_15_r20/external/pdfium/third_party/agg23/
H A Dagg_rendering_buffer.h79 int8u* row_ptr = m_buf; in attach() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DFlashAttentionKernel.cpp315 accum_t* row_ptr = qk_data + row * kvBlockSize; in cpu_flash_attention() local
608 accum_t* row_ptr = attn_data + row * kvBlockSize; in cpu_flash_attention_backward() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DFormatting.cpp192 const double *row_ptr = row.const_data_ptr<double>(); in __printMatrix() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cpp33 const auto* row_ptr = sizes_data + row_num * sizes_row_length; in padded_tensor_numel() local
/aosp_15_r20/external/executorch/extension/llm/custom_ops/
H A Dop_sdpa.cpp492 accum_t* row_ptr = qk_data + row * kvBlockSize; in cpu_flash_attention() local
/aosp_15_r20/external/deqp/external/openglcts/modules/gl/
H A Dgl4cTextureViewTests.cpp4669 … const unsigned char *row_ptr = m_result_data + m_reference_texture_width * y * 4 /* RGBA */; in executeTest() local
8447 const unsigned char *row_ptr = data_ptr + y * m_texture_n_components * write_to_width; in checkProgramWriteCoherency() local
H A Dgl4cShaderSubroutineTests.cpp7445 const float *row_ptr = (const float *)m_read_buffer + y * m_to_width * 4; /* rgba */ in verifyReadBuffer() local
9372 const float *row_ptr = m_to_data + y * 4 /* rgba */ * m_to_width; in verifyRenderedData() local
H A Dgl4cGPUShaderFP64Tests.cpp9275 const unsigned char *row_ptr = m_to_data + 4 /* rgba */ * m_to_width * y; in executeFunctionalTest() local
/aosp_15_r20/hardware/google/graphics/common/libhwc2.1/libhwchelper/
DExynosHWCHelper.cpp1394 png_bytep row_ptr = (png_bytep)bufferHandleData; in load_png_image() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected.cc754 const int8_t* row_ptr = GetTensorData<int8_t>(filter); in EvalHybrid() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReduceOps.cpp1788 const char* row_ptr = data[0] + outer_stride * i; in std_var_all_cpu() local