Home
last modified time | relevance | path

Searched refs:row_ptr (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dtranspose_op.cc57 const int num_rows = input.row_ptr.size() - 1; in ValidateTransposeInputs()
65 if (num_cols + 1 != output.row_ptr.size()) { in ValidateTransposeInputs()
68 " vs. ", output.row_ptr.size()); in ValidateTransposeInputs()
224 const int num_rows = input.row_ptr.size() - 1; in operator ()()
231 output->row_ptr(input.col_ind(i) + 1) += 1; in operator ()()
233 std::partial_sum(output->row_ptr.data(), in operator ()()
234 output->row_ptr.data() + num_cols + 1, in operator ()()
235 output->row_ptr.data()); in operator ()()
241 const int64 row_begin = input.row_ptr(row_idx); in operator ()()
242 const int64 row_end = input.row_ptr(row_idx + 1); in operator ()()
[all …]
Dadd_op.cc278 const int m = a.row_ptr.size() - 1; in GetOutputStructure()
279 DCHECK_EQ(m, b.row_ptr.size() - 1); in GetOutputStructure()
291 m, n, descrA_.descr(), nnzA, a.row_ptr.data(), a.col_ind.data(), in GetOutputStructure()
292 descrB_.descr(), nnzB, b.row_ptr.data(), b.col_ind.data(), in GetOutputStructure()
306 const int m = a.row_ptr.size() - 1; in Compute()
307 DCHECK_EQ(m, b.row_ptr.size() - 1); in Compute()
319 m, n, &alpha_, descrA_.descr(), nnzA, a.values.data(), a.row_ptr.data(), in Compute()
321 b.row_ptr.data(), b.col_ind.data(), descrC_.descr(), c->values.data(), in Compute()
322 c->row_ptr.data(), c->col_ind.data())); in Compute()
Dsparse_mat_mul_op.cc571 DCHECK_EQ(m, a.row_ptr.size() - 1); in GetOutputStructure()
577 DCHECK_EQ(k, b.row_ptr.size() - 1); in GetOutputStructure()
588 transA_, transB_, m, n, k, descrA_.descr(), nnzA, a.row_ptr.data(), in GetOutputStructure()
589 a.col_ind.data(), descrB_.descr(), nnzB, b.row_ptr.data(), in GetOutputStructure()
607 DCHECK_EQ(m, a.row_ptr.size() - 1); in Compute()
610 DCHECK_EQ(m, c->row_ptr.size() - 1); in Compute()
614 DCHECK_EQ(k, b.row_ptr.size() - 1); in Compute()
625 a.row_ptr.data(), a.col_ind.data(), descrB_.descr(), nnzB, in Compute()
626 b.values.data(), b.row_ptr.data(), b.col_ind.data(), descrC_.descr(), in Compute()
627 c->values.data(), c->row_ptr.data(), c->col_ind.data())); in Compute()
Dkernels_gpu.cu.cc384 const int* row_ptr, in CSRSparseMatrixSoftmaxKernel2D() argument
391 CalculateRowSoftmax(ldg(row_ptr + row), ldg(row_ptr + row + 1), logits, in CSRSparseMatrixSoftmaxKernel2D()
410 const int* row_ptr, const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel3D() argument
425 CalculateRowSoftmax(batch_offset + ldg(row_ptr + row_offset), in CSRSparseMatrixSoftmaxKernel3D()
426 batch_offset + ldg(row_ptr + row_offset + 1), logits, in CSRSparseMatrixSoftmaxKernel3D()
437 auto row_ptr = logits.row_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGPUImpl() local
445 DCHECK_EQ(rows, row_ptr.size() - 1); in CSRSparseMatrixSoftmaxGPUImpl()
449 d.stream(), rows /*size*/, row_ptr.data(), in CSRSparseMatrixSoftmaxGPUImpl()
455 DCHECK_EQ((rows + 1) * batch_size, row_ptr.size()); in CSRSparseMatrixSoftmaxGPUImpl()
471 batch_ptr_copy.data(), row_ptr.data(), in CSRSparseMatrixSoftmaxGPUImpl()
Dcsr_sparse_matrix_to_dense_op.cc77 auto row_ptr = csr_sparse_matrix->row_pointers().vec<int32>(); in Compute() local
102 const int64 col_begin = row_ptr(row_offset); in Compute()
103 const int64 col_end = row_ptr(row_offset + 1); in Compute()
Dsparse_matrix.h618 TTypes<int32>::UnalignedConstVec row_ptr; member
626 TTypes<int32>::UnalignedVec row_ptr; member
Dmat_mul_op.cc801 a.values.data(), a.row_ptr.data(), a.col_ind.data(), in Compute()
855 a.values.data(), a.row_ptr.data(), in Compute()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.cc68 const int8_t* row_ptr = matrix + row * m_cols; in SseMatrixBatchVectorMultiplyAccumulate() local
78 _mm_loadu_si128(reinterpret_cast<const __m128i*>(row_ptr + col)); in SseMatrixBatchVectorMultiplyAccumulate()
88 sum += row_ptr[col] * vectors[col]; in SseMatrixBatchVectorMultiplyAccumulate()
107 const int8_t* row_ptr = matrix + row * m_cols; in SseMatrixBatchVectorMultiplyAccumulate() local
115 _mm_loadu_si128(reinterpret_cast<const __m128i*>(row_ptr + col)); in SseMatrixBatchVectorMultiplyAccumulate()
132 sum += row_ptr[col] * vectors[col]; in SseMatrixBatchVectorMultiplyAccumulate()
133 row_sum += row_ptr[col]; in SseMatrixBatchVectorMultiplyAccumulate()
153 const int8_t* row_ptr = matrix; in SseSparseMatrixBatchVectorMultiplyAccumulate() local
163 _mm_loadu_si128(reinterpret_cast<const __m128i*>(row_ptr)); in SseSparseMatrixBatchVectorMultiplyAccumulate()
166 row_ptr += kBlockSize; in SseSparseMatrixBatchVectorMultiplyAccumulate()
Dneon_tensor_utils.cc756 int8_t* row_ptr = in NeonMatrixBatchVectorMultiplyImpl() local
759 memcpy(aligned_row, row_ptr, sizeof(int8_t) * n_input); in NeonMatrixBatchVectorMultiplyImpl()
760 row_ptr = aligned_row; in NeonMatrixBatchVectorMultiplyImpl()
773 (uintptr_t)(&row_ptr[col]) & (kWeightsPerUint32 - 1), 0); in NeonMatrixBatchVectorMultiplyImpl()
775 const int8x16_t s2_8x16 = vld1q_s8((const int8_t*)(row_ptr + col)); in NeonMatrixBatchVectorMultiplyImpl()
799 (uintptr_t)(&row_ptr[col]) & (kWeightsPerUint32 - 1), 0); in NeonMatrixBatchVectorMultiplyImpl()
801 const int8x8_t s2_8x8 = vld1_s8((const int8_t*)(row_ptr + col)); in NeonMatrixBatchVectorMultiplyImpl()
812 dotprod += row_ptr[col] * aligned_vec[col]; in NeonMatrixBatchVectorMultiplyImpl()
1079 int8_t* row_ptr = (int8_t*)matrix + row * m_cols; // NOLINT in NeonMatrixBatchVectorMultiplyAccumulate() local
1081 memcpy(aligned_row, row_ptr, sizeof(int8_t) * m_cols); in NeonMatrixBatchVectorMultiplyAccumulate()
[all …]
Ddepthwiseconv_3x3_filter_common.h585 const T* row_ptr = input_block_data;
587 const T* ptr = row_ptr;
593 row_ptr += input_height_stride;
/external/libpng/contrib/oss-fuzz/
Dlibpng_read_fuzzer.cc30 if (png_handler.row_ptr) \
31 png_free(png_handler.png_ptr, png_handler.row_ptr); \
41 png_handler.row_ptr = nullptr; \
55 png_voidp row_ptr = nullptr; member
59 if (row_ptr) in ~PngObjectHandler()
60 png_free(png_ptr, row_ptr); in ~PngObjectHandler()
99 png_handler.row_ptr = nullptr; in LLVMFuzzerTestOneInput()
175 png_handler.row_ptr = png_malloc( in LLVMFuzzerTestOneInput()
182 static_cast<png_bytep>(png_handler.row_ptr), nullptr); in LLVMFuzzerTestOneInput()
/external/pdfium/third_party/agg23/
Dagg_rendering_buffer.h77 int8u* row_ptr = m_buf; in attach() local
79 row_ptr = m_buf - int(height - 1) * stride; in attach()
83 *rows++ = row_ptr; in attach()
84 row_ptr += stride; in attach()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.cc147 const int8_t* row_ptr = matrix; in PortableMatrixBatchVectorMultiplyAccumulate() local
153 __builtin_prefetch(row_ptr, 0 /* prefetch for read */, in PortableMatrixBatchVectorMultiplyAccumulate()
156 for (int col = 0; col < m_cols; ++col, ++row_ptr) { in PortableMatrixBatchVectorMultiplyAccumulate()
157 dotprod += (*row_ptr) * (vectors[col]); in PortableMatrixBatchVectorMultiplyAccumulate()
172 const int8_t* row_ptr = matrix; in PortableMatrixBatchVectorMultiplyAccumulate() local
177 __builtin_prefetch(row_ptr, 0 /* prefetch for read */, in PortableMatrixBatchVectorMultiplyAccumulate()
180 for (int col = 0; col < m_cols; ++col, ++row_ptr) { in PortableMatrixBatchVectorMultiplyAccumulate()
181 dotprod += (*row_ptr) * (vectors[col] - batch_offset); in PortableMatrixBatchVectorMultiplyAccumulate()
231 const int8_t* row_ptr = matrix; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local
237 __builtin_prefetch(row_ptr, 0 /* prefetch for read */, in PortableSparseMatrixBatchVectorMultiplyAccumulate()
[all …]
/external/deqp/external/openglcts/modules/gles31/
Des31cTextureStorageMultisampleFunctionalTests.cpp667 float* row_ptr = (float*)buffer + row_width * data_y; in iterate() local
671 float* data_ptr = row_ptr + x * pixel_size; in iterate()
1054 const unsigned int* row_ptr = (unsigned int*)((char*)buffer + data_y * row_width); in iterate() local
1058 const unsigned int* data_ptr = (unsigned int*)((char*)row_ptr + x * pixel_size); in iterate()
/external/deqp/external/openglcts/modules/gl/
Dgl4cTextureViewTests.cpp4672 const unsigned char* row_ptr = m_result_data + m_reference_texture_width * y * 4 /* RGBA */; in executeTest() local
4676 const unsigned char* pixel_ptr = row_ptr + x * 4 /* RGBA */; in executeTest()
8438 const unsigned char* row_ptr = data_ptr + y * m_texture_n_components * write_to_width; in checkProgramWriteCoherency() local
8442 const unsigned char* texel_ptr = row_ptr + x * m_texture_n_components; in checkProgramWriteCoherency()
Dgl4cShaderSubroutineTests.cpp7430 const float* row_ptr = (const float*)m_read_buffer + y * m_to_width * 4; /* rgba */ in verifyReadBuffer() local
7434 const float* texel_ptr = row_ptr + x * 4; /* rgba */ in verifyReadBuffer()
9347 const float* row_ptr = m_to_data + y * 4 /* rgba */ * m_to_width; in verifyRenderedData() local
9351 const float* pixel_ptr = row_ptr + 4 /* rgba */ * x; in verifyRenderedData()
Dgl4cGPUShaderFP64Tests.cpp9214 const unsigned char* row_ptr = m_to_data + 4 /* rgba */ * m_to_width * y; in executeFunctionalTest() local
9218 const unsigned char* pixel_ptr = row_ptr + 4 /* rgba */ * x; in executeFunctionalTest()
/external/libyuv/files/source/
Drow_mmi.cc5701 : [row_ptr] "r"(row), [pre_ptr] "r"(previous_cumsum), in ComputeCumulativeSumRow_MMI()
/external/libpng/
DCHANGES6003 Initialize png_handler.row_ptr in contrib/oss-fuzz/libpng_read_fuzzer.cc
6036 Initialize png_handler.row_ptr in contrib/oss-fuzz/libpng_read_fuzzer.cc