Home
last modified time | relevance | path

Searched defs:scaling_factors (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc131 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/2); in Prepare() local
180 TfLiteTensor* scaling_factors, in EvalHybrid()
240 TfLiteTensor* scaling_factors = GetTemporary(context, node, 2); in Eval() local
Dunidirectional_sequence_rnn.cc132 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/2); in Prepare() local
208 TfLiteTensor* hidden_state_scratch, TfLiteTensor* scaling_factors, in EvalHybrid()
310 TfLiteTensor* scaling_factors = GetTemporary(context, node, 2); in Eval() local
Dsvdf.cc217 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/2); in Prepare() local
289 TfLiteTensor* scratch, TfLiteTensor* scaling_factors, in EvalHybrid()
389 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/2); in Eval() local
Dfully_connected.cc152 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/1); in Prepare() local
210 TfLiteTensor* scaling_factors, TfLiteTensor* output) { in EvalHybrid()
343 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/1); in EvalQuantized() local
Dunidirectional_sequence_lstm.cc390 TfLiteTensor* scaling_factors = in Prepare() local
527 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/4); in Eval() local
Dbidirectional_sequence_rnn.cc231 TfLiteTensor* scaling_factors = in Prepare() local
423 TfLiteTensor* scaling_factors, TfLiteTensor* input_quantized, in EvalHybrid()
662 TfLiteTensor* scaling_factors = in Eval() local
Dlstm.cc461 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/4); in Prepare() local
606 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/4); in Eval() local
Dbidirectional_sequence_lstm.cc683 TfLiteTensor* scaling_factors = in Prepare() local
965 TfLiteTensor* scaling_factors = in Eval() local
Dlstm_eval.cc430 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
1068 TfLiteStatus EvalHybrid( in EvalHybrid()
Dconv.cc399 TfLiteTensor* scaling_factors = in Prepare() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc126 int8_t* quantized_hidden_state_ptr_batch, float* scaling_factors, in RnnBatchStep()
149 int8_t* quantized_hidden_state_ptr_batch, float* scaling_factors, in RnnBatchStep()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.h37 const int8_t* __restrict__ vectors, const float* scaling_factors, in MatrixBatchVectorMultiplyAccumulate()
54 const float* scaling_factors, int n_batch, float* __restrict__ result, in SparseMatrixBatchVectorMultiplyAccumulate()
Dneon_tensor_utils.cc198 const int8_t* vectors, const float* scaling_factors, int n_batch, in DotprodMatrixBatchFourVectorMultiplyAccumulate()
316 const float* scaling_factors, int n_batch, float* __restrict__ result, in DotprodSparseMatrixBatchVectorMultiplyAccumulate()
381 const int8_t* __restrict__ vectors, const float* scaling_factors, in NeonMatrixBatchVectorMultiplyAccumulate()
556 const float* scaling_factors, int n_batch, float* __restrict__ result, in NeonSparseMatrixBatchVectorMultiplyAccumulate()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h171 const int8_t* __restrict__ vector, const float* scaling_factors, in MatrixBatchVectorMultiplyAccumulate()
188 const float* scaling_factors, int n_batch, float* __restrict__ result, in SparseMatrixBatchVectorMultiplyAccumulate()
Dportable_tensor_utils.cc90 const int8_t* __restrict__ vectors, const float* scaling_factors, in PortableMatrixBatchVectorMultiplyAccumulate()
146 const float* scaling_factors, int n_batch, float* __restrict__ result, in PortableSparseMatrixBatchVectorMultiplyAccumulate()