get_tensor_format - 2.5 日本語

ランナーのテンソル形式を取得します。

プロトタイプ

TensorFormat get_tensor_format();

パラメーター

なし

戻り値

TensorFormat: NHWC/HCHW

使用法

auto format = runner->get_tensor_format();
switch (format) {
    case vart::Runner::TensorFormat::NCHW:
        // do something
        break;
    case vart::Runner::TensorFormat::NHWC:
        // do something
        break;
}