Home
last modified time | relevance | path

Searched refs:batch_vector (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_common.h369 const T* batch_vector, int n_batch, in VectorBatchVectorCwiseProduct() argument
372 VectorVectorCwiseProduct(vector, batch_vector, v_size, result); in VectorBatchVectorCwiseProduct()
375 batch_vector += v_size; in VectorBatchVectorCwiseProduct()
384 const T* batch_vector, in VectorBatchVectorCwiseProductAccumulate() argument
387 VectorVectorCwiseProductAccumulate(vector, batch_vector, v_size, result); in VectorBatchVectorCwiseProductAccumulate()
390 batch_vector += v_size; in VectorBatchVectorCwiseProductAccumulate()
396 const int16_t* batch_vector,
403 T* batch_vector) { in VectorBatchVectorAdd() argument
406 batch_vector[i] += vector[i]; in VectorBatchVectorAdd()
408 batch_vector += v_size; in VectorBatchVectorAdd()
[all …]
Dtensor_utils_test.cc1751 const std::vector<int16_t> batch_vector = { in TEST() local
1779 batch_vector.data(), kBatchSize, in TEST()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils_impl.h105 const int16_t* vector, int v_size, const int16_t* batch_vector, int n_batch,
186 int n_batch, float* batch_vector);
Dportable_tensor_utils.h246 const int16_t* batch_vector, in VectorBatchVectorCwiseProductAccumulate() argument
250 vector, v_size, batch_vector, n_batch, multiplier, shift, result); in VectorBatchVectorCwiseProductAccumulate()
Dportable_tensor_utils.cc684 const int16_t* vector, int v_size, const int16_t* batch_vector, int n_batch, in PortableVectorBatchVectorCwiseProductAccumulate() argument
688 int32_t prod = vector[v] * *batch_vector++; in PortableVectorBatchVectorCwiseProductAccumulate()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h229 const int16_t* batch_vector, in VectorBatchVectorCwiseProductAccumulate() argument
233 batch_vector, n_batch, multiplier, shift, result); in VectorBatchVectorCwiseProductAccumulate()
Dneon_tensor_utils.h219 const int16_t* batch_vector, in VectorBatchVectorCwiseProductAccumulate() argument
223 batch_vector, n_batch, multiplier, shift, result); in VectorBatchVectorCwiseProductAccumulate()
Dneon_tensor_utils_impl.h172 const int16_t* vector, int v_size, const int16_t* batch_vector, int n_batch,
Dneon_tensor_utils.cc2518 const int16_t* vector, int v_size, const int16_t* batch_vector, int n_batch, in NeonVectorBatchVectorCwiseProductAccumulate() argument
2527 prod.val[0] = vmull_s16(vld1_s16(vector + v), vld1_s16(batch_vector)); in NeonVectorBatchVectorCwiseProductAccumulate()
2529 vmull_s16(vld1_s16(vector + v + 4), vld1_s16(batch_vector + 4)); in NeonVectorBatchVectorCwiseProductAccumulate()
2531 vmull_s16(vld1_s16(vector + v + 8), vld1_s16(batch_vector + 8)); in NeonVectorBatchVectorCwiseProductAccumulate()
2533 vmull_s16(vld1_s16(vector + v + 12), vld1_s16(batch_vector + 12)); in NeonVectorBatchVectorCwiseProductAccumulate()
2534 batch_vector += 16; in NeonVectorBatchVectorCwiseProductAccumulate()
2568 int32_t prod = vector[v] * *batch_vector++; in NeonVectorBatchVectorCwiseProductAccumulate()