Home
last modified time | relevance | path

Searched refs:vector_block_in_batch_ptr (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dportable_tensor_utils.cc219 const float* vector_block_in_batch_ptr = in PortableSparseMatrixBatchVectorMultiplyAccumulate1x4() local
222 dot_prod += *matrix_ptr++ * *vector_block_in_batch_ptr++; in PortableSparseMatrixBatchVectorMultiplyAccumulate1x4()
247 const int8_t* vector_block_in_batch_ptr = in PortableSparseMatrixBatchVectorMultiplyAccumulate1x16() local
250 dot_prod += *matrix_ptr * *vector_block_in_batch_ptr++; in PortableSparseMatrixBatchVectorMultiplyAccumulate1x16()
282 const float* vector_block_in_batch_ptr = in PortableSparseMatrixBatchVectorMultiplyAccumulate() local
285 dot_prod += *matrix_ptr++ * *vector_block_in_batch_ptr++; in PortableSparseMatrixBatchVectorMultiplyAccumulate()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dneon_tensor_utils.cc1944 const float* vector_block_in_batch_ptr = in NeonSparseMatrixBatchVectorMultiplyAccumulate1x4() local
1948 float32x4_t vector_f32x4 = vld1q_f32(vector_block_in_batch_ptr); in NeonSparseMatrixBatchVectorMultiplyAccumulate1x4()
1980 const int8_t* vector_block_in_batch_ptr = in NeonSparseMatrixBatchVectorMultiplyAccumulate1x16() local
1984 int8x16_t vector_i8x16 = vld1q_s8(vector_block_in_batch_ptr); in NeonSparseMatrixBatchVectorMultiplyAccumulate1x16()
2055 const float* vector_block_in_batch_ptr = in NeonSparseMatrixBatchVectorMultiplyAccumulate() local
2060 float32x4_t vector_f32x4 = vld1q_f32(vector_block_in_batch_ptr + in NeonSparseMatrixBatchVectorMultiplyAccumulate()