if not kLinear it needs to be something like https://github.com/NVIDIA/TensorRT/blob/main/samples/common/buffers.h#L286
if not kLinear it needs to be something like https://github.com/NVIDIA/TensorRT/blob/main/samples/common/buffers.h#L286