Skip to content

Add support for aten::prelu.Tensor #150

Closed
@xsacha

Description

@xsacha

aten::prelu.Tensor

Appearance in TorchScript:
class PReLU(Module):
parameters = ["weight", ]
weight : Tensor
training : bool
def forward(self: torch.torch.nn.modules.activation.PReLU,
argument_1: Tensor) -> Tensor:
input = torch.prelu(argument_1, self.weight)
return input

Original PyTorch API: implicit PyTorch Tensor op

Relevant TensorRT Documentation:

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions