Searched defs:LogSoftmax (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | log_softmax.h | 28 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
60 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
245 inline void LogSoftmax(const SoftmaxParams& params, const size_t outer_size, in LogSoftmax() function
|
| D | legacy_reference_ops.h | 769 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
776 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
1791 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1797 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
|
| /external/armnn/src/backends/reference/workloads/ |
| D | LogSoftmax.cpp | 29 void LogSoftmax(Decoder<float>& input, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | log_softmax.h | 25 inline void LogSoftmax(int32_t input_multiplier, int32_t input_shift, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations_test.cc | 2278 TEST_P(LogSoftmaxOpTest, LogSoftmax) { in TEST_P() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4231 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4238 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
4244 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4259 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
|
| D | optimized_ops.h | 3754 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
3777 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
3800 inline void LogSoftmax(const SoftmaxParams& params, float input_scale, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3186 TEST_F(OpTest, LogSoftmax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 22868 func LogSoftmax(scope *Scope, logits tf.Output) (logsoftmax tf.Output) { func
|