Home
last modified time | relevance | path

Searched defs:leaky_relu (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dleaky-relu.c145 pytorch_qnnp_operator_t leaky_relu, in pytorch_qnnp_setup_leaky_relu_nc_q8()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dactivations.py514 leaky_relu = nn.leaky_relu variable
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/modules/
H A Dfused.py205 def __init__(self, linear, leaky_relu): argument
/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A Donednn.py76 def _fuse_linear_bn_leaky_relu(is_qat, linear, bn, leaky_relu): argument
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dactivation.h146 inline Tensor leaky_relu(Tensor input, double negative_slope, bool inplace) { in leaky_relu() function
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py556 def leaky_relu( function
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops_fulljit.cpp738 at::Tensor leaky_relu(const at::Tensor& tensor, double scalar) { in leaky_relu() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h889 auto leaky_relu = in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/pytorch/torch/_refs/nn/functional/
H A D__init__.py366 def leaky_relu( function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DClamp.cpp558 Tensor leaky_relu(const Tensor& self_arg, const Scalar& negative_slope) { in leaky_relu() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DActivation.cpp165 TORCH_META_FUNC(leaky_relu) ( in TORCH_META_FUNC() argument
/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dsubgraph.h246 } leaky_relu; member
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py3664 def leaky_relu(features, alpha=0.2, name=None): function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py1882 def leaky_relu( function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py1259 def leaky_relu( function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp2962 TEST_F(VulkanAPITest, leaky_relu) { in TEST_F() argument