Searched refs:SoftmaxParams (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | softmax_common.cc | 34 SoftmaxParams* op_data) { in CalculateSoftmaxParams()
93 return context->AllocatePersistentBuffer(context, sizeof(SoftmaxParams)); in SoftmaxInit()
106 SoftmaxParams* op_data = static_cast<SoftmaxParams*>(node->user_data); in SoftmaxPrepare()
|
| D | softmax.cc | 32 const SoftmaxParams& op_data) { in SoftmaxQuantized()
67 SoftmaxParams op_data = *static_cast<SoftmaxParams*>(node->user_data); in SoftmaxEval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | softmax.cc | 38 SoftmaxParams params;
279 SoftmaxParams op_data = *static_cast<SoftmaxParams*>(node->user_data); in Eval()
293 SoftmaxParams op_data = *static_cast<SoftmaxParams*>(node->user_data); in Eval()
301 SoftmaxParams op_data = *static_cast<SoftmaxParams*>(node->user_data); in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | softmax.cc | 32 const SoftmaxParams& op_data) { in SoftmaxQuantized()
74 const SoftmaxParams data = in SoftmaxEval()
75 *static_cast<const SoftmaxParams*>(node->user_data); in SoftmaxEval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 30 inline void Softmax(const SoftmaxParams& params, in Softmax()
67 inline void Softmax(const SoftmaxParams& params, in Softmax()
148 inline int16_t SoftMaxCalculateExp(const SoftmaxParams& params, in SoftMaxCalculateExp()
165 inline void SoftmaxInt16(const SoftmaxParams& params, in SoftmaxInt16()
|
| D | legacy_reference_ops.h | 751 SoftmaxParams params; in Softmax()
760 SoftmaxParams params; in Softmax()
769 SoftmaxParams params; in LogSoftmax()
779 SoftmaxParams params; in LogSoftmax()
|
| D | reference_ops.h | 905 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
937 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | softmax_test.cc | 97 SoftmaxParams params; in TEST()
126 SoftmaxParams params; in TEST()
|
| D | log_softmax_test.cc | 94 SoftmaxParams params; in TEST()
|
| D | activations.cc | 72 struct SoftmaxParams params = {};
90 struct SoftmaxParams params = {};
1052 SoftmaxParams op_params; in SoftmaxFloat()
1190 SoftmaxParams op_params; in LogSoftmaxEval()
1203 SoftmaxParams op_params = data->params; in LogSoftmaxEval()
1218 SoftmaxParams op_params = data->params; in LogSoftmaxEval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | logsoftmax_quantized_test.cc | 53 SoftmaxParams sm_params; in RunLogSoftmaxFloatReference()
87 SoftmaxParams sm_params; in RunLogSoftmaxFloatReference()
173 SoftmaxParams params; in RunOneLogSoftmaxTest()
|
| D | softmax_quantized_test.cc | 51 SoftmaxParams sm_params; in RunSoftmaxFloatReference()
128 SoftmaxParams params; in RunOneSoftmaxTest()
|
| D | types.h | 1078 struct SoftmaxParams { struct
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3647 inline void SoftmaxImpl(const SoftmaxParams& params, in SoftmaxImpl()
3673 SoftmaxWorkerTask(const SoftmaxParams& params, in SoftmaxWorkerTask()
3690 const tflite::SoftmaxParams& params;
3699 inline void Softmax(const SoftmaxParams& params,
3753 inline void PopulateSoftmaxLookupTable(SoftmaxParams* data, float input_scale, in PopulateSoftmaxLookupTable()
3763 inline void Softmax(const SoftmaxParams& params, in Softmax()
3837 inline void PopulateSoftmaxUInt8LookupTable(SoftmaxParams* data, in PopulateSoftmaxUInt8LookupTable()
3904 inline void SoftmaxInt8LUT(const SoftmaxParams& params, in SoftmaxInt8LUT()
4068 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
4091 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
[all …]
|
| D | legacy_optimized_ops.h | 3990 inline void Softmax(const SoftmaxParams& params, in Softmax()
4195 SoftmaxParams params; in Softmax()
4211 SoftmaxParams params; in Softmax()
4228 SoftmaxParams params; in LogSoftmax()
4244 SoftmaxParams params; in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 1177 struct SoftmaxParams {
|