Function tvm::tirx::transform::FlattenTensor

• Defined in File transform.h

Function Documentation

Pass tvm::tirx::transform::FlattenTensor()

Flatten the multi-dimensional TensorLoad and TensorStore to single dimensional TensorLoad/TensorStore for the TIR not contains opaque block.

Returns:

The pass.