| /aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ops/ |
| H A D | MetalLeakyReLU.mm | 16 static Tensor& leaky_relu_(Tensor& input, const Scalar& negative_slope_val) {
85 m.impl(TORCH_SELECTIVE_NAME("aten::leaky_relu_"), TORCH_FN(leaky_relu_));
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Clamp.cpp | 564 Tensor& leaky_relu_(Tensor& self, const Scalar& negative_slope) { in leaky_relu_() function
567 return ops::activation_scalar_(self, scalar, VK_KERNEL(leaky_relu_)); in leaky_relu_()
610 m.impl(TORCH_SELECTIVE_NAME("aten::leaky_relu_"), leaky_relu_); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/tools/pyi/ |
| H A D | gen_pyi.py | 414 "leaky_relu_": [
415 … f"def leaky_relu_({', '.join(['input: Tensor', 'negative_slope: float = ...'])}) -> Tensor: ..."
518 "leaky_relu_",
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | nn.functional.rst | 85 leaky_relu_
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_x86inductor_quantizer.py | 1285 nn.LeakyReLU, aten.leaky_relu.default, aten.leaky_relu_.default
1298 nn.LeakyReLU, aten.leaky_relu.default, aten.leaky_relu_.default, is_qat=True
1313 aten.leaky_relu_.default,
1329 aten.leaky_relu_.default,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/ |
| H A D | tf_op_names.inc | 291 StringAttr leaky_relu_;
566 leaky_relu_ = StringAttr::get(getContext(), "tfg.LeakyRelu");
|
| H A D | tf_op_names.cc | 460 return op_name == leaky_relu_; in IsLeakyRelu()
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 337 - aten::leaky_relu_
856 aten::leaky_relu_: 5
|
| H A D | tensor_ops.py | 90 a.leaky_relu_(),
|
| H A D | model_ops.yaml | 199 aten::leaky_relu_: 4
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 892 auto leaky_relu_ = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
893 "aten::leaky_relu_", {"%negative_slope"}); in quant_fusion_pattern_and_replacements()
1072 std::move(leaky_relu_), in quant_fusion_pattern_and_replacements()
|
| H A D | helper.cpp | 164 "leaky_relu_",
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | interface.cpp | 61 aten::leaky_relu_, in fuseGraph()
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| H A D | dcgan_training.txt | 19 Operator: aten.leaky_relu_.default
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/ |
| H A D | executorch.py | 364 F.leaky_relu_,
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 343 aten.leaky_relu_,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/ |
| H A D | MPSCNNTests.mm | 360 auto Y1 = at::leaky_relu_(X, -0.0125);
361 auto Y2 = at::leaky_relu_(X2, -0.0125).cpu();
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | top_operators_github_usage.py | 341 ("nn.functional.leaky_relu_", 68),
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | discover_coverage.py | 249 "nn.functional.leaky_relu_",
|
| /aosp_15_r20/external/executorch/backends/qualcomm/quantizer/ |
| H A D | annotators.py | 337 torch.ops.aten.leaky_relu_.default,
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/ |
| H A D | functional.py | 586 result = torch._C._nn.leaky_relu_(input, negative_slope)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | allowlist_for_publicAPI.json | 602 "leaky_relu_",
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 1900 result = torch._C._nn.leaky_relu_(input, negative_slope)
1906 leaky_relu_ = _add_docstr( variable
1907 torch._C._nn.leaky_relu_,
1909 leaky_relu_(input, negative_slope=0.01) -> Tensor
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | activation.h | 148 return torch::leaky_relu_(input, negative_slope); in leaky_relu()
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/timm_train/ |
| H A D | cspdarknet53_training.txt | 92 Operator: aten.leaky_relu_.default
|