Skip to content

🐛 Check for the casting of the unary operations in conversion/impl/unary/ops.py #3241

@apbose

Description

@apbose

The following lines-

    if (isinstance(input_val, TRTTensor)) and (
        input_val.dtype == trt.int8 or input_val.dtype == trt.int32
    ):
        input_val = cast_trt_tensor(ctx, input_val, trt.float32, name)

are present in almost all the unary ops. Check its requirement

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions