Home
last modified time | relevance | path

Searched refs:exp_lut (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsoftmax.cc34 uint16_t* exp_lut; member
86 input_diff == 0 ? kMaxExponentValue : op_data.exp_lut[input_diff]; in SoftmaxHifimini()
97 input_diff == 0 ? kMaxExponentValue : op_data.exp_lut[input_diff]; in SoftmaxHifimini()
151 op_data->exp_lut[i] = static_cast<uint16_t>(exponent_scaled); in CalculateSoftmaxOpDataHifimini()
173 op_data->exp_lut = static_cast<uint16_t*>(context->AllocatePersistentBuffer( in PrepareHifimini()
175 TF_LITE_ENSURE(context, op_data->exp_lut != nullptr); in PrepareHifimini()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsoftmax_common.cc112 op_data->exp_lut = reinterpret_cast<int16_t*>(raw_exp_lut); in SoftmaxPrepare()
134 op_data->exp_lut, kInt16LUTArraySize); in SoftmaxPrepare()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsoftmax.h162 return generic_int16_table_lookup(sat_sym_scaled_diff, params.exp_lut); in SoftMaxCalculateExp()
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc79 int16_t exp_lut[kInt16LUTArraySize]; // int16 LUT for exp(x), where x uniform member
630 data->params.exp_lut = data->exp_lut; in SoftmaxPrepare()
634 data->params.exp_lut, data->kInt16LUTArraySize); in SoftmaxPrepare()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h100 int16_t* exp_lut; member
1191 int16_t* exp_lut;
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h1093 int16_t* exp_lut; member