| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | softmax_test.cc | 92 TEST(Softmax, DISABLED_Beta) { in TEST() argument
117 TEST(Softmax, MultiThreading) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | softmax_test.cc | 30 TEST_F(OpenCLOperationTest, Softmax) { in TEST_F() argument
|
| /external/armnn/src/backends/reference/workloads/ |
| D | Softmax.cpp | 17 void Softmax(Decoder<float>& in, Encoder<float>& out, const TensorInfo& inputTensorInfo, float beta… in Softmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 30 inline void Softmax(const SoftmaxParams& params, in Softmax() function
67 inline void Softmax(const SoftmaxParams& params, in Softmax() function
|
| D | legacy_reference_ops.h | 748 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
756 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
1773 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
1780 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | softmax.cc | 42 class Softmax : public NodeShader { class
|
| D | softmax_test.cc | 35 TEST(SoftmaxTest, Softmax) { in TEST() argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | advanced_activations.py | 295 class Softmax(Layer): class
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3993 inline void Softmax(const SoftmaxParams& params, in Softmax() function
4195 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
4203 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
4210 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
4220 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
|
| D | optimized_ops.h | 3772 inline void Softmax(const SoftmaxParams& params, in Softmax() function
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference.cc | 821 void Softmax(const int32_t rank, in Softmax() function
|
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 820 void Softmax(const int32_t rank, in Softmax() function
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 820 void Softmax(const int32_t rank, in Softmax() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 737 class Softmax class
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2902 TEST_F(OpTest, Softmax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 9046 func Softmax(scope *Scope, logits tf.Output) (softmax tf.Output) { func
|