Home
last modified time | relevance | path

Searched defs:vector_in_batch (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.cc128 const float* vector_in_batch = vector + b * m_cols; in PortableMatrixBatchVectorMultiplyAccumulate() local
216 const float* vector_in_batch = vector + batch * m_cols; in PortableSparseMatrixBatchVectorMultiplyAccumulate1x4() local
244 const int8_t* vector_in_batch = vector + batch * m_cols; in PortableSparseMatrixBatchVectorMultiplyAccumulate1x16() local
279 const float* vector_in_batch = vector + batch * m_cols; in PortableSparseMatrixBatchVectorMultiplyAccumulate() local
Dsvdf.h99 const int8_t* vector_in_batch = input_data + b * n_input; in EvalIntegerSVDF() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.cc170 const float* vector_in_batch = vector + b * m_cols; in Avx2MatrixBatchVectorMultiplyAccumulateImpl() local
Dneon_tensor_utils.cc236 const float* vector_in_batch = vector + b * m_cols; in NeonMatrixBatchVectorMultiplyAccumulate() local
1940 const float* vector_in_batch = vector + batch * m_cols; in NeonSparseMatrixBatchVectorMultiplyAccumulate1x4() local
1976 const int8_t* vector_in_batch = vector + batch * m_cols; in NeonSparseMatrixBatchVectorMultiplyAccumulate1x16() local
2051 const float* vector_in_batch = vector + batch * m_cols; in NeonSparseMatrixBatchVectorMultiplyAccumulate() local