Searched refs:SoftmaxParams (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax.cc | 96 SoftmaxParams op_params; in Softmax1DQuantized()
115 SoftmaxParams op_params; in Softmax2DQuantized()
127 SoftmaxParams op_params; in Softmax4DFloat()
136 SoftmaxParams op_params; in Softmax4DQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 601 SoftmaxParams op_params; in Softmax3DFloat()
617 SoftmaxParams op_params; in Softmax1DQuantizedUint8()
634 SoftmaxParams op_params; in Softmax2DQuantizedUint8()
650 SoftmaxParams op_params; in Softmax3DQuantizedUint8()
664 SoftmaxParams op_params; in Softmax4DFloat()
673 SoftmaxParams op_params; in Softmax4DQuantizedUint8()
687 SoftmaxParams op_params; in Softmax1DQuantizedInt8()
701 SoftmaxParams op_params; in Softmax2DQuantizedInt8()
717 SoftmaxParams op_params; in Softmax3DQuantizedInt8()
730 SoftmaxParams op_params; in Softmax4DQuantizedInt8()
[all …]
|
| D | softmax_test.cc | 96 SoftmaxParams params; in TEST()
125 SoftmaxParams params; in TEST()
|
| D | log_softmax_test.cc | 94 SoftmaxParams params; in TEST()
|
| /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()
123 SoftmaxParams params; in RunOneSoftmaxTest()
|
| D | types.h | 951 struct SoftmaxParams { struct
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 28 inline void Softmax(const SoftmaxParams& params, in Softmax()
60 inline void Softmax(const SoftmaxParams& params, in Softmax()
|
| D | legacy_reference_ops.h | 701 SoftmaxParams params; in Softmax()
710 SoftmaxParams params; in Softmax()
719 SoftmaxParams params; in LogSoftmax()
729 SoftmaxParams params; in LogSoftmax()
|
| D | reference_ops.h | 2567 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
2599 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 24 inline void Softmax(const SoftmaxParams& params, in Softmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1430 SoftmaxParams params; in Softmax()
1446 SoftmaxParams params; in Softmax()
1463 SoftmaxParams params; in LogSoftmax()
1479 SoftmaxParams params; in LogSoftmax()
|
| D | optimized_ops.h | 4431 inline void Softmax(const SoftmaxParams& params, in Softmax()
4451 inline void Softmax(const SoftmaxParams& params, in Softmax()
4664 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
4700 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 906 struct SoftmaxParams { struct
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4546 inline void Softmax(const SoftmaxParams& params, in Softmax()
4571 SoftmaxParams params; in Softmax()
4576 inline void Softmax(const SoftmaxParams& params, in Softmax()
4793 SoftmaxParams params; in Softmax()
4802 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
4841 SoftmaxParams params; in LogSoftmax()
4960 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax()
5057 SoftmaxParams params; in LogSoftmax()
|