Searched refs:kBlockSize (Results 1 – 4 of 4) sorted by relevance
295 static const std::intptr_t kBlockSize = 16; in SseSparseMatrixVectorMultiplyAccumulate() local296 TFLITE_DCHECK_EQ(m_cols % kBlockSize, 0); in SseSparseMatrixVectorMultiplyAccumulate()303 const std::intptr_t col_index = *ledger_ptr++ * kBlockSize; in SseSparseMatrixVectorMultiplyAccumulate()311 matrix += kBlockSize; in SseSparseMatrixVectorMultiplyAccumulate()329 static const std::intptr_t kBlockSize = 16; in SseSparseMatrix4VectorsMultiplyAccumulate() local330 TFLITE_DCHECK_EQ(m_cols % kBlockSize, 0); in SseSparseMatrix4VectorsMultiplyAccumulate()350 const std::intptr_t col_index = *ledger++ * kBlockSize; in SseSparseMatrix4VectorsMultiplyAccumulate()368 matrix += kBlockSize; in SseSparseMatrix4VectorsMultiplyAccumulate()420 static constexpr std::intptr_t kBlockSize = 16; in SseReductionSumVector() local425 for (; col < (reduction_size & ~(kBlockSize - 1)); col += kBlockSize) { in SseReductionSumVector()
1908 constexpr int kBlockSize = kFloatValuesPerNeonVector; in NeonSparseMatrixBatchVectorMultiplyAccumulate1x4() local1909 TFLITE_DCHECK_EQ(m_cols % kBlockSize, 0); in NeonSparseMatrixBatchVectorMultiplyAccumulate1x4()1927 matrix_ptr += kBlockSize; in NeonSparseMatrixBatchVectorMultiplyAccumulate1x4()1941 m_cols % kBlockSize, 0); in NeonSparseMatrixBatchVectorMultiplyAccumulate()1966 matrix_ptr += kBlockSize; in NeonSparseMatrixBatchVectorMultiplyAccumulate()1987 constexpr int kBlockSize = kInt8ValuesPerNeonVector; in NeonSparseMatrixBatchVectorMultiplyAccumulate() local1989 m_cols % kBlockSize, 0); in NeonSparseMatrixBatchVectorMultiplyAccumulate()2034 row_ptr += kBlockSize; in NeonSparseMatrixBatchVectorMultiplyAccumulate()2587 for (; i <= v_size - kBlockSize; i += kBlockSize) { in NeonMeanStddevNormalization()2617 for (; i <= v_size - kBlockSize; i += kBlockSize) { in NeonMeanStddevNormalization()[all …]
210 const int kBlockSize = 4; in PortableSparseMatrixBatchVectorMultiplyAccumulate1x4() local211 TFLITE_DCHECK_EQ(m_cols % kBlockSize, 0); in PortableSparseMatrixBatchVectorMultiplyAccumulate1x4()218 const int block_start_index = indices[i] * kBlockSize; in PortableSparseMatrixBatchVectorMultiplyAccumulate1x4()221 for (int c = 0; c < kBlockSize; c++) { in PortableSparseMatrixBatchVectorMultiplyAccumulate1x4()234 const int kBlockSize = 16; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local236 m_cols % kBlockSize, 0); in PortableSparseMatrixBatchVectorMultiplyAccumulate()249 for (int c = 0; c < kBlockSize; c++) { in PortableSparseMatrixBatchVectorMultiplyAccumulate()263 static const int kBlockSize = 16; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local265 m_cols % kBlockSize, 0); in PortableSparseMatrixBatchVectorMultiplyAccumulate()281 const int block_start_index = *ledger_ptr++ * kBlockSize; in PortableSparseMatrixBatchVectorMultiplyAccumulate()[all …]
37 static const size_t kBlockSize = 16384; // default initial size in ImgIoUtilReadFromStdin() local50 const size_t extra_size = (max_size == 0) ? kBlockSize : max_size; in ImgIoUtilReadFromStdin()
Completed in 18 milliseconds