/external/tensorflow/tensorflow/lite/kernels/ |
D | basic_rnn.cc | 131 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
|
D | unidirectional_sequence_rnn.cc | 132 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
|
D | svdf.cc | 217 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
|
D | fully_connected.cc | 152 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
|
D | unidirectional_sequence_lstm.cc | 390 TfLiteTensor* scaling_factors = in Prepare() local 527 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/4); in Eval() local
|
D | bidirectional_sequence_rnn.cc | 231 TfLiteTensor* scaling_factors = in Prepare() local 423 TfLiteTensor* scaling_factors, TfLiteTensor* input_quantized, in EvalHybrid() 662 TfLiteTensor* scaling_factors = in Eval() local
|
D | lstm.cc | 461 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/4); in Prepare() local 606 TfLiteTensor* scaling_factors = GetTemporary(context, node, /*index=*/4); in Eval() local
|
D | bidirectional_sequence_lstm.cc | 683 TfLiteTensor* scaling_factors = in Prepare() local 965 TfLiteTensor* scaling_factors = in Eval() local
|
D | lstm_eval.cc | 430 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput() 1068 TfLiteStatus EvalHybrid( in EvalHybrid()
|
D | conv.cc | 399 TfLiteTensor* scaling_factors = in Prepare() local
|
/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | kernel_utils.cc | 126 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/ |
D | neon_tensor_utils.h | 37 const int8_t* __restrict__ vectors, const float* scaling_factors, in MatrixBatchVectorMultiplyAccumulate() 54 const float* scaling_factors, int n_batch, float* __restrict__ result, in SparseMatrixBatchVectorMultiplyAccumulate()
|
D | neon_tensor_utils.cc | 198 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/ |
D | portable_tensor_utils.h | 171 const int8_t* __restrict__ vector, const float* scaling_factors, in MatrixBatchVectorMultiplyAccumulate() 188 const float* scaling_factors, int n_batch, float* __restrict__ result, in SparseMatrixBatchVectorMultiplyAccumulate()
|
D | portable_tensor_utils.cc | 90 const int8_t* __restrict__ vectors, const float* scaling_factors, in PortableMatrixBatchVectorMultiplyAccumulate() 146 const float* scaling_factors, int n_batch, float* __restrict__ result, in PortableSparseMatrixBatchVectorMultiplyAccumulate()
|