Home
last modified time | relevance | path

Searched refs:TrivialTestGraphInputYielder (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.h29 class TrivialTestGraphInputYielder : public InputYielder {
31 TrivialTestGraphInputYielder(int num_stages, int width, int tensor_size,
Dtrivial_test_graph_input_yielder.cc89 TrivialTestGraphInputYielder::TrivialTestGraphInputYielder( in TrivialTestGraphInputYielder() function in tensorflow::grappler::TrivialTestGraphInputYielder
98 bool TrivialTestGraphInputYielder::NextItem(GrapplerItem* item) { in NextItem()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory_test.cc46 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"/CPU:0"}); in TEST_F()
74 TrivialTestGraphInputYielder fake_input(4, 1, -1, false, {"/CPU:0"}); in TEST_F()
100 TrivialTestGraphInputYielder fake_input(4, 2, 1024 * 1024, false, in TEST_F()
138 TrivialTestGraphInputYielder fake_input(4, 2, 1024 * 1024, false, {"/GPU:0"}); in TEST_F()
Dgraph_properties_test.cc123 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()
167 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()
194 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()
1659 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc136 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
155 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
176 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
199 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
216 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
715 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
737 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
756 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
774 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
Dstatic_schedule_test.cc50 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
126 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
Dimplementation_selector_test.cc44 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {CpuDevice}); in TEST_F()
Dmodel_pruner_test.cc37 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
Ddependency_optimizer_test.cc53 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
Dloop_optimizer_test.cc518 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
Darithmetic_optimizer_test.cc92 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_test.cc30 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {{"CPU:0"}}); in TEST_F()
Dgraph_view_test.cc127 TrivialTestGraphInputYielder fake_input(4, 2, 2, false, {"/CPU:0", "/GPU:0"}); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc64 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()
Dsingle_machine_test.cc68 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()
101 TrivialTestGraphInputYielder fake_input(4, 1, 10, true, in TEST_F()
112 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()
/external/tensorflow/tensorflow/core/grappler/utils/
Dscc_test.cc56 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F()