| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 28 inline void Softmax(const SoftmaxParams& params, in Softmax() function
60 inline void Softmax(const SoftmaxParams& params, in Softmax() function
135 inline void Softmax(const float* in, const int input_size, const int batch_size, in Softmax() function
|
| D | legacy_reference_ops.h | 698 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
706 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
1677 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
1684 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 24 inline void Softmax(const SoftmaxParams& params, in Softmax() function
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | advanced_activations.py | 239 class Softmax(Layer): class
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1427 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
1435 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
1442 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
1452 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
|
| D | optimized_ops.h | 4431 inline void Softmax(const SoftmaxParams& params, in Softmax() function
4451 inline void Softmax(const SoftmaxParams& params, in Softmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 549 void Softmax(const float* in, const int input_size, const int batch_size, in Softmax() function
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4546 inline void Softmax(const SoftmaxParams& params, in Softmax() function
4568 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
4576 inline void Softmax(const SoftmaxParams& params, in Softmax() function
4789 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 882 class Softmax class
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2901 TEST_F(OpTest, Softmax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 8281 func Softmax(scope *Scope, logits tf.Output) (softmax tf.Output) { func
|