| 33 | namespace { |
| 34 | |
| 35 | inline Graph LoadGraph(const Device& device) { |
| 36 | Graph graph = |
| 37 | Graph::Load("./oneflow/api/cpp/tests/graph_test_model/affine_with_parameter", device); |
| 38 | return graph; |
| 39 | } |
| 40 | |
| 41 | inline void Forward(Graph& graph, const Device& device, int expected_batch_dim = 1) { |
| 42 | std::vector<float> data(expected_batch_dim * 3); |