Home
last modified time | relevance | path

Searched defs:mish (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/fuser/
H A Drun_benchmarks.py166 def mish(a): function
/aosp_15_r20/external/pytorch/functorch/benchmarks/
H A Dpointwise_scorecard.py153 def mish(a): function
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_memory_efficient_fusion.py34 def mish(x): function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dactivation.h393 inline Tensor mish(const Tensor& input) { in mish() function
/aosp_15_r20/external/pytorch/torch/_refs/nn/functional/
H A D__init__.py390 def mish(a: TensorLikeType, inplace: bool = False) -> TensorLikeType: function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DActivation.cpp146 TORCH_META_FUNC(mish) (const Tensor& self) { in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py2383 def mish(input: Tensor, inplace: bool = False) -> Tensor: function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py1211 def mish(g: jit_utils.GraphContext, input): function