Home
last modified time | relevance | path

Searched defs:Softmax (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dsoftmax_test.cc92 TEST(Softmax, DISABLED_Beta) { in TEST() argument
117 TEST(Softmax, MultiThreading) { in TEST() argument
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dsoftmax_test.cc30 TEST_F(OpenCLOperationTest, Softmax) { in TEST_F() argument
/external/armnn/src/backends/reference/workloads/
DSoftmax.cpp17 void Softmax(Decoder<float>& in, Encoder<float>& out, const TensorInfo& inputTensorInfo, float beta… in Softmax() function
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsoftmax.h31 inline void Softmax(const SoftmaxParams& params, in Softmax() function
68 inline void Softmax(const SoftmaxParams& params, in Softmax() function
Dlegacy_reference_ops.h750 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
758 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
1775 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
1782 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dsoftmax.cc43 class Softmax : public NodeShader { class
Dsoftmax_test.cc35 TEST(SoftmaxTest, Softmax) { in TEST() argument
/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
Dsoftmax_op_benchmark.cc34 std::string Softmax(llvm::ArrayRef<bool> dynamic_dims, in Softmax() function
/external/tensorflow/tensorflow/python/keras/layers/
Dadvanced_activations.py295 class Softmax(Layer): class
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3995 inline void Softmax(const SoftmaxParams& params, in Softmax() function
4197 inline void Softmax(const float* input_data, const RuntimeShape& input_shape, in Softmax() function
4205 inline void Softmax(const float* input_data, const Dims<4>& input_dims, in Softmax() function
4212 inline void Softmax(const uint8* input_data, const RuntimeShape& input_shape, in Softmax() function
4222 inline void Softmax(const uint8* input_data, const Dims<4>& input_dims, in Softmax() function
Doptimized_ops.h3447 inline void Softmax(const SoftmaxParams& params, in Softmax() function
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference_beta.cc820 void Softmax(const int32_t rank, in Softmax() function
Donedevice_train_palm_detection_filter_inference.cc821 void Softmax(const int32_t rank, in Softmax() function
Donedevice_train_palm_detection_filter_inference_v2.cc820 void Softmax(const int32_t rank, in Softmax() function
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc741 class Softmax class
/external/ComputeLibrary/
Dfilelist.json758 "Softmax": { object
2109 "Softmax": { object
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc4145 TEST_F(OpTest, Softmax) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go43675 func Softmax(scope *Scope, logits tf.Output) (softmax tf.Output) { func