Searched defs:LogSoftmax (Results 1 – 8 of 8) sorted by relevance
| /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/internal/reference/ |
| 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
1784 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
1790 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
|
| D | reference_ops.h | 905 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
937 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations_test.cc | 2131 TEST(FloatActivationsOpTest, LogSoftmax) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4226 inline void LogSoftmax(const float* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4233 inline void LogSoftmax(const float* input_data, const Dims<4>& input_dims, in LogSoftmax() function
4239 inline void LogSoftmax(const uint8* input_data, const RuntimeShape& input_shape, in LogSoftmax() function
4254 inline void LogSoftmax(const uint8* input_data, const Dims<4>& input_dims, in LogSoftmax() function
|
| D | optimized_ops.h | 4068 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4091 inline void LogSoftmax(const SoftmaxParams& params, in LogSoftmax() function
4114 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 | 24709 func LogSoftmax(scope *Scope, logits tf.Output) (logsoftmax tf.Output) { func
|