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 | 767 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
774 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
1789 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1795 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 | 23 inline void LogSoftmax(int32_t input_multiplier, int32_t input_shift, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations_test.cc | 2196 TEST_P(LogSoftmaxOpTest, LogSoftmax) { in TEST_P() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4229 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4236 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
4242 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4257 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
|
| D | optimized_ops.h | 4077 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4100 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4123 inline void LogSoftmax(const SoftmaxParams& params, float input_scale, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2192 TEST_F(OpTest, LogSoftmax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 9022 func LogSoftmax(scope *Scope, logits tf.Output) (logsoftmax tf.Output) { func
|