Method get_value

imperative/src/impl/interpreter/interpreter_impl.cpp:487–508

}

HostTensorND ChannelImpl::get_value(Handle handle) {
    MGB_LOCK_GUARD(m_spin);
    assert_available();
    mgb_assert(
            m_valid_handle.find(handle) != m_valid_handle.end(), "invalid handle: %p",
            handle);
    auto info = reinterpret_cast<TensorInfo*>(handle);
    // donnot use info->value_fetched, it's unsafe
    mgb_assert(!info->invalid, "tensor is unusable due to previous error");

    // pin
    SmallVector<TensorInfo*> vec({info});
    m_dtr.pin(vec);

    auto ret = wait_tensor(info, TensorProp::HostValue)->get_value();

    // unpin
    auto& state = get_channel_state();
    auto dtr_evictee_minimum_size = state.options.dtr_evictee_minimum_size;
    m_dtr.unpin(vec, dtr_evictee_minimum_size);
    return ret;
}

TensorShape ChannelImpl::get_shape(Handle handle) {
    MGB_LOCK_GUARD(m_spin);