Home
last modified time | relevance | path

Searched defs:batch_vector (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_common.h369 const T* batch_vector, int n_batch, in VectorBatchVectorCwiseProduct()
384 const T* batch_vector, in VectorBatchVectorCwiseProductAccumulate()
403 T* batch_vector) { in VectorBatchVectorAdd()
415 T* batch_vector) { in VectorBatchVectorAssign()
Dtensor_utils_test.cc1751 const std::vector<int16_t> batch_vector = { in TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h246 const int16_t* batch_vector, in VectorBatchVectorCwiseProductAccumulate()
Dportable_tensor_utils.cc684 const int16_t* vector, int v_size, const int16_t* batch_vector, int n_batch, in PortableVectorBatchVectorCwiseProductAccumulate()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h229 const int16_t* batch_vector, in VectorBatchVectorCwiseProductAccumulate()
Dneon_tensor_utils.h219 const int16_t* batch_vector, in VectorBatchVectorCwiseProductAccumulate()
Dneon_tensor_utils.cc2518 const int16_t* vector, int v_size, const int16_t* batch_vector, int n_batch, in NeonVectorBatchVectorCwiseProductAccumulate()