| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | op_log_softmax.py | 19 class LogSoftmax(NodeVisitor): class
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | LogSoftmax.cpp | 29 void LogSoftmax(Decoder<float>& input, in LogSoftmax() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| H A D | log_softmax.h | 25 inline void LogSoftmax(int32_t input_multiplier, int32_t input_shift, in LogSoftmax() function
|
| /aosp_15_r20/external/executorch/backends/arm/test/ops/ |
| H A D | test_logsoftmax.py | 34 class LogSoftmax(torch.nn.Module): class in TestLogSoftmax
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | activation.py | 1703 class LogSoftmax(Module): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | activations_test.cc | 2278 TEST_P(LogSoftmaxOpTest, LogSoftmax) { in TEST_P() argument
|
| /aosp_15_r20/external/executorch/backends/qualcomm/tests/ |
| H A D | models.py | 609 class LogSoftmax(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | functional.cpp | 1392 TEST_F(FunctionalTest, LogSoftmax) { in TEST_F() argument
|
| H A D | modules.cpp | 2615 TEST_F(ModulesTest, LogSoftmax) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 3186 TEST_F(OpTest, LogSoftmax) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
| H A D | wrappers.go | 22868 func LogSoftmax(scope *Scope, logits tf.Output) (logsoftmax tf.Output) { func
|