Home
last modified time | relevance | path

Searched refs:ApplyActivationToVector (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc71 tensor_utils::ApplyActivationToVector( in RnnBatchStep()
110 tensor_utils::ApplyActivationToVector( in RnnBatchStep()
248 tensor_utils::ApplyActivationToVector( in RnnBatchStep()
330 tensor_utils::ApplyActivationToVector( in RnnBatchStep()
Dtensor_utils.h192 inline void ApplyActivationToVector(const float* __restrict__ vector, in ApplyActivationToVector() function
Dtensor_utils_test.cc1709 ApplyActivationToVector(input, kVectorSize, kTfLiteActRelu, output.data()); in TEST()
1713 ApplyActivationToVector(input, kVectorSize, kTfLiteActTanh, output.data()); in TEST()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsvdf.h62 tensor_utils::ApplyActivationToVector(output_ptr, batch_size * num_units, in ApplyTimeWeightsBiasAndActivation()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc252 tensor_utils::ApplyActivationToVector(cell_scratch, n_batch * n_cell, in LstmStepWithAuxInput()
287 tensor_utils::ApplyActivationToVector(cell_state_ptr, n_batch * n_cell, in LstmStepWithAuxInput()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc94 tensor_utils::ApplyActivationToVector(gate, n_batch * n_cell, activation, in CalculateLstmGateFloat()
129 tensor_utils::ApplyActivationToVector(cell_state, n_batch * n_cell, in CalculateLstmOutputCalibration()
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc427 tensor_utils::ApplyActivationToVector( in EvalPie()
475 tensor_utils::ApplyActivationToVector( in EvalHybridImpl()
520 tensor_utils::ApplyActivationToVector(per_thread_output, in EvalHybridImpl()
Dlstm_eval.cc199 tensor_utils::ApplyActivationToVector(gate, n_batch * n_cell, activation, in CalculateLstmGateFloat()
269 tensor_utils::ApplyActivationToVector(cell_state, n_batch * n_cell, in CalculateLstmOutputFloat()
410 tensor_utils::ApplyActivationToVector(gate, n_cell * n_batch, activation, in CalculateLstmGateHybrid()
441 tensor_utils::ApplyActivationToVector(cell_state, n_batch * n_cell, in CalculateLstmOutputHybrid()