| /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/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/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
1768 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
1775 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | advanced_activations.py | 290 class Softmax(Layer): class
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3990 inline void Softmax(const SoftmaxParams& params, in Softmax() function
4192 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
4200 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
4207 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
4217 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
|
| D | optimized_ops.h | 3763 inline void Softmax(const SoftmaxParams& params, in Softmax() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 755 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 | 24733 func Softmax(scope *Scope, logits tf.Output) (softmax tf.Output) { func
|