Gets the tensor format of runner.
Prototype
TensorFormat get_tensor_format();
Parameters
None
Returns
TensorFormat: NHWC / HCHW
Usage
auto format = runner->get_tensor_format();
switch (format) {
case vart::Runner::TensorFormat::NCHW:
// do something
break;
case vart::Runner::TensorFormat::NHWC:
// do something
break;
}