Разница в параметрах графа и библиотеки после relay.build_module.build () в библиотеке TVM - PullRequest
Новая
       16

Разница в параметрах графа и библиотеки после relay.build_module.build () в библиотеке TVM

0 голосов
/ 18 июня 2019

В библиотеке TVM, когда используется _relay.build_module.build()_, каков результат вывода после relay.build и tvm.build

И какова разница и потребность в двух параметрах в качестве вывода - графики библиотека - которые позже используются для запуска графа или создания графика времени выполнения?

Например, в этом коде: 

opt_level = 3
target = tvm.target.cuda()
with relay.build_config(opt_level=opt_level):
    graph, lib, params = relay.build_module.build(
        net, target, params=params)

Каковы выходные данные graph , lib ?Можем ли мы использовать только один из них для генерации времени выполнения графика, или мы всегда должны использовать оба из них, как показано ниже: 

# create random input
ctx = tvm.gpu()
data = np.random.uniform(-1, 1, size=data_shape).astype("float32")
# create module
module = graph_runtime.create(graph, lib, ctx)
# set input and parameters
module.set_input("data", data)
module.set_input(**params)
# run
module.run()
# get output
out = module.get_output(0, tvm.nd.empty(out_shape)).asnumpy()
...