| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | log_softmax.h | 23 inline void LogSoftmax(int32_t input_multiplier, int32_t input_shift, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations_test.cc | 758 TEST(FloatActivationsOpTest, LogSoftmax) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1461 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
1468 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1474 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
1488 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
|
| D | optimized_ops.h | 4664 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4700 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 717 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
724 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
1693 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1699 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
|
| D | reference_ops.h | 2567 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
2599 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1830 class LogSoftmax : public SimpleOperator<LogSoftmaxOperator> { class
1832 explicit LogSoftmax() in LogSoftmax() function in toco::tflite::LogSoftmax
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4802 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4839 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4960 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
5052 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2191 TEST_F(OpTest, LogSoftmax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 8257 func LogSoftmax(scope *Scope, logits tf.Output) (logsoftmax tf.Output) { func
|