get_tensor_format - 2.0 English

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;
}