Searched refs:NodeSetUp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 55 NodeSetUp("Node1", kConv2D, kCPU0, 6000, &node1_); in VirtualSchedulerTest()
56 NodeSetUp("Node2", kConv2D, kCPU0, 5000, &node2_); in VirtualSchedulerTest()
57 NodeSetUp("Node3", kConv2D, kCPU0, 4000, &node3_); in VirtualSchedulerTest()
58 NodeSetUp("Node4", kConv2D, kCPU0, 3000, &node4_); in VirtualSchedulerTest()
59 NodeSetUp("Node5", kConv2D, kCPU0, 2000, &node5_); in VirtualSchedulerTest()
60 NodeSetUp("Node6", kConv2D, kCPU0, 1000, &node6_); in VirtualSchedulerTest()
103 void NodeSetUp(const string& name, const string& op_name, in NodeSetUp() function in tensorflow::grappler::VirtualSchedulerTest
1952 NodeSetUp("Node7", kConv2D, kCPU0, 5, &node7); in TEST_F()
1953 NodeSetUp("Node8", kConv2D, kCPU0, 4, &node8); in TEST_F()
1954 NodeSetUp("Node9", kConv2D, kCPU0, 3, &node9); in TEST_F()
[all …]
|