Function tvm::script::ir_builder::tirx::Arg(ffi::String, TensorVar)

• Defined in File ir.h

Function Documentation

TensorVar tvm::script::ir_builder::tirx::Arg(ffi::String name, TensorVar tensor)

The Function tensor arguments adding function.

Parameters:

• name – The name of the tensor.

• tensor – The tensor argument.

Returns:

The tensor.