| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | transpose_op.cc | 57 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 …]
|
| D | sparse_mat_mul_op.cc | 608 DCHECK_EQ(m, a.row_ptr.size() - 1); in CSRSparseSparseMatrixMatMul()
613 DCHECK_EQ(k, b.row_ptr.size() - 1); in CSRSparseSparseMatrixMatMul()
622 m, n, k, descrA_.descr(), nnzA, a.row_ptr.data(), a.col_ind.data(), in CSRSparseSparseMatrixMatMul()
623 descrB_.descr(), nnzB, b.row_ptr.data(), b.col_ind.data(), info_, in CSRSparseSparseMatrixMatMul()
639 DCHECK_EQ(m, a.row_ptr.size() - 1); in CSRSparseSparseMatrixMatMul()
645 DCHECK_EQ(k, b.row_ptr.size() - 1); in CSRSparseSparseMatrixMatMul()
657 transA_, transB_, m, n, k, descrA_.descr(), nnzA, a.row_ptr.data(), in CSRSparseSparseMatrixMatMul()
658 a.col_ind.data(), descrB_.descr(), nnzB, b.row_ptr.data(), in CSRSparseSparseMatrixMatMul()
662 m, n, k, descrA_.descr(), nnzA, a.row_ptr.data(), a.col_ind.data(), in CSRSparseSparseMatrixMatMul()
663 descrB_.descr(), nnzB, b.row_ptr.data(), b.col_ind.data(), in CSRSparseSparseMatrixMatMul()
[all …]
|
| D | add_op.cc | 296 const int m = a.row_ptr.size() - 1; in GetWorkspaceSize()
297 DCHECK_EQ(m, b.row_ptr.size() - 1); in GetWorkspaceSize()
310 m, n, &alpha_, descrA_.descr(), nnzA, a.values.data(), a.row_ptr.data(), in GetWorkspaceSize()
312 b.row_ptr.data(), b.col_ind.data(), descrC_.descr(), null_T, null_int, in GetWorkspaceSize()
324 const int m = a.row_ptr.size() - 1; in GetOutputStructure()
325 DCHECK_EQ(m, b.row_ptr.size() - 1); in GetOutputStructure()
337 m, n, descrA_.descr(), nnzA, a.row_ptr.data(), a.col_ind.data(), in GetOutputStructure()
338 descrB_.descr(), nnzB, b.row_ptr.data(), b.col_ind.data(), in GetOutputStructure()
352 const int m = a.row_ptr.size() - 1; in Compute()
353 DCHECK_EQ(m, b.row_ptr.size() - 1); in Compute()
[all …]
|
| D | kernels_gpu.cu.cc | 371 const int* row_ptr, in CSRSparseMatrixSoftmaxKernel2D() argument
378 CalculateRowSoftmax(ldg(row_ptr + row), ldg(row_ptr + row + 1), logits, in CSRSparseMatrixSoftmaxKernel2D()
397 const int* row_ptr, const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel3D() argument
412 CalculateRowSoftmax(batch_offset + ldg(row_ptr + row_offset), in CSRSparseMatrixSoftmaxKernel3D()
413 batch_offset + ldg(row_ptr + row_offset + 1), logits, in CSRSparseMatrixSoftmaxKernel3D()
424 auto row_ptr = logits.row_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGPUImpl() local
432 DCHECK_EQ(rows, row_ptr.size() - 1); in CSRSparseMatrixSoftmaxGPUImpl()
436 d.stream(), rows /*size*/, row_ptr.data(), in CSRSparseMatrixSoftmaxGPUImpl()
442 DCHECK_EQ((rows + 1) * batch_size, row_ptr.size()); in CSRSparseMatrixSoftmaxGPUImpl()
458 batch_ptr_copy.data(), row_ptr.data(), in CSRSparseMatrixSoftmaxGPUImpl()
|
| D | csr_sparse_matrix_to_dense_op.cc | 77 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()
|
| D | mat_mul_op.cc | 840 &matA, m, k, nnz, const_cast<int*>(a.row_ptr.data()), in Compute()
898 a.values.data(), a.row_ptr.data(), a.col_ind.data(), in Compute()
956 a.values.data(), a.row_ptr.data(), in Compute()
960 a.values.data(), a.row_ptr.data(), in Compute()
|
| D | sparse_matrix.h | 618 TTypes<int32>::UnalignedConstVec row_ptr; member
626 TTypes<int32>::UnalignedVec row_ptr; member
|
| /external/libpng/contrib/oss-fuzz/ |
| D | libpng_read_fuzzer.cc | 30 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/ |
| D | agg_rendering_buffer.h | 77 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/optimized/ |
| D | sse_tensor_utils.cc | 108 const int8_t* __restrict__ row_ptr = matrix + row * m_cols; in SseMatrixBatchVectorMultiplyAccumulateImpl() local
122 _mm_loadu_si128(reinterpret_cast<const __m128i*>(row_ptr + col)); in SseMatrixBatchVectorMultiplyAccumulateImpl()
134 _mm_loadl_epi64(reinterpret_cast<const __m128i*>(row_ptr + col))); in SseMatrixBatchVectorMultiplyAccumulateImpl()
145 _mm_loadl_epi64(reinterpret_cast<const __m128i*>(row_ptr + col))); in SseMatrixBatchVectorMultiplyAccumulateImpl()
163 sum += row_ptr[col] * vectors[col]; in SseMatrixBatchVectorMultiplyAccumulateImpl()
421 const int8_t* __restrict__ row_ptr = input_vector + row * reduction_size; in SseReductionSumVector() local
426 _mm_loadu_si128(reinterpret_cast<const __m128i*>(row_ptr + col)); in SseReductionSumVector()
435 _mm_loadl_epi64(reinterpret_cast<const __m128i*>(row_ptr + col))); in SseReductionSumVector()
449 row_sum += row_ptr[col]; in SseReductionSumVector()
|
| D | neon_tensor_utils.cc | 784 int8_t* row_ptr = in NeonMatrixBatchVectorMultiplyImpl() local
787 memcpy(aligned_row, row_ptr, sizeof(int8_t) * n_input); in NeonMatrixBatchVectorMultiplyImpl()
788 row_ptr = aligned_row; in NeonMatrixBatchVectorMultiplyImpl()
801 (uintptr_t)(&row_ptr[col]) & (kNeonVectorAlignment - 1), 0); in NeonMatrixBatchVectorMultiplyImpl()
803 const int8x16_t s2_8x16 = vld1q_s8((const int8_t*)(row_ptr + col)); in NeonMatrixBatchVectorMultiplyImpl()
826 (uintptr_t)(&row_ptr[col]) & (kNeonVectorAlignment - 1), 0); in NeonMatrixBatchVectorMultiplyImpl()
828 const int8x8_t s2_8x8 = vld1_s8((const int8_t*)(row_ptr + col)); in NeonMatrixBatchVectorMultiplyImpl()
838 dotprod += row_ptr[col] * aligned_vec[col]; in NeonMatrixBatchVectorMultiplyImpl()
1105 int8_t* row_ptr = (int8_t*)matrix + row * m_cols; // NOLINT in NeonMatrixBatchVectorMultiplyAccumulate() local
1107 memcpy(aligned_row, row_ptr, sizeof(int8_t) * m_cols); in NeonMatrixBatchVectorMultiplyAccumulate()
[all …]
|
| D | depthwiseconv_3x3_filter_common.h | 572 const T* row_ptr = input_block_data;
574 const T* ptr = row_ptr;
580 row_ptr += input_height_stride;
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.cc | 145 const int8_t* row_ptr = matrix; in PortableMatrixBatchVectorMultiplyAccumulate() local
151 __builtin_prefetch(row_ptr, 0 /* prefetch for read */, in PortableMatrixBatchVectorMultiplyAccumulate()
154 for (int col = 0; col < m_cols; ++col, ++row_ptr) { in PortableMatrixBatchVectorMultiplyAccumulate()
155 dotprod += (*row_ptr) * (vectors[col]); in PortableMatrixBatchVectorMultiplyAccumulate()
184 const int8_t* row_ptr = matrix; in PortableMatrixBatchVectorMultiplyAccumulate() local
193 __builtin_prefetch(row_ptr, 0 /* prefetch for read */, in PortableMatrixBatchVectorMultiplyAccumulate()
196 for (int col = 0; col < m_cols; ++col, ++row_ptr) { in PortableMatrixBatchVectorMultiplyAccumulate()
197 dotprod += (*row_ptr) * vectors[col]; in PortableMatrixBatchVectorMultiplyAccumulate()
270 const int8_t* row_ptr = matrix; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local
276 __builtin_prefetch(row_ptr, 0 /* prefetch for read */, in PortableSparseMatrixBatchVectorMultiplyAccumulate()
[all …]
|
| /external/deqp/external/openglcts/modules/gles31/ |
| D | es31cTextureStorageMultisampleFunctionalTests.cpp | 667 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/ |
| D | gl4cTextureViewTests.cpp | 4672 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()
|
| D | gl4cShaderSubroutineTests.cpp | 7430 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()
|
| D | gl4cGPUShaderFP64Tests.cpp | 9214 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/ |
| D | row_mmi.cc | 5701 : [row_ptr] "r"(row), [pre_ptr] "r"(previous_cumsum), in ComputeCumulativeSumRow_MMI()
|
| /external/libpng/ |
| D | CHANGES | 6003 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
|