MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / TEST

Function TEST

oneflow/api/cpp/tests/graph_test.cpp:60–65  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

58} // namespace
59
60TEST(Api, graph_cpu_test) {
61 EnvScope scope;
62 Device device("cpu");
63 Graph graph = LoadGraph(device);
64 Forward(graph, device, 1);
65}
66
67#ifdef WITH_CUDA
68TEST(Api, graph_gpu_test) {

Callers

nothing calls this directly

Calls 15

LoadGraphFunction · 0.85
ForwardFunction · 0.85
threadClass · 0.85
bindFunction · 0.85
joinMethod · 0.80
copy_toMethod · 0.80
GetInputInfosMethod · 0.80
GetOutputInfosMethod · 0.80
ShapeClass · 0.50
set_batch_sizeMethod · 0.45
emplace_backMethod · 0.45
beginMethod · 0.45

Tested by

no test coverage detected