Searched defs:add_relu (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/ao/nn/quantized/modules/ |
| D | functional_modules.py | 88 def add_relu(self, x: Tensor, y: Tensor) -> Tensor: member in FloatFunctional
153 def add_relu(self, x: Tensor, y: Tensor) -> Tensor: member in FXFloatFunctional
277 def add_relu(self, x: Tensor, y: Tensor) -> Tensor: member in QFunctional
|
| /external/pytorch/torch/ao/ns/ |
| D | _numeric_suite.py | 332 def add_relu(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: member in Shadow
|
| /external/pytorch/torch/csrc/jit/passes/quantization/ |
| D | quantization_patterns.h | 450 std::string add_relu = R"( in quant_fusion_pattern_and_replacements() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | BinaryOps.cpp | 847 Tensor add_relu(const Tensor& self, const Tensor& other, const Scalar& alpha) { in add_relu() function
852 Tensor add_relu(const Tensor& self, const Scalar& other, const Scalar& alpha) { in add_relu() function
|