Home
last modified time | relevance | path

Searched defs:Softplus (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsoftplus_op.h32 struct Softplus { struct
60 // Functor used by SoftplusGradOp to do the computations. argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dunary_ops.cc96 static xla::XlaOp Softplus(xla::XlaBuilder* b, xla::XlaOp features) { in Softplus() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/
H A Dgpu_unary_ops_test.cc1147 GENERATE_DEFAULT_TEST_2(Softplus, DT_HALF, DT_FLOAT, DT_HALF, DT_FLOAT, in GENERATE_DEFAULT_TEST_2() argument
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dactivation.py858 class Softplus(Module): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp1785 TEST_F(FunctionalTest, Softplus) { in TEST_F() argument
H A Dmodules.cpp2953 TEST_F(ModulesTest, Softplus) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc4165 TEST_F(OpTest, Softplus) { in TEST_F() argument