Searched refs:ApplySigmoidToVector (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils.h | 183 inline void ApplySigmoidToVector(const float* __restrict__ vector, int v_size, in ApplySigmoidToVector() function
210 return ApplySigmoidToVector(vector, v_size, result); in ApplyActivationToVector()
|
| D | tensor_utils_test.cc | 1700 ApplySigmoidToVector(input, kVectorSize, output.data()); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 214 tensor_utils::ApplySigmoidToVector(input_gate_scratch, n_cell * n_batch, in LstmStepWithAuxInput()
235 tensor_utils::ApplySigmoidToVector(forget_gate_scratch, n_cell * n_batch, in LstmStepWithAuxInput()
285 tensor_utils::ApplySigmoidToVector(output_gate_scratch, n_batch * n_cell, in LstmStepWithAuxInput()
|