Home
last modified time | relevance | path

Searched defs:fake_input (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory_test.cc46 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"/CPU:0"}); in TEST_F() local
74 TrivialTestGraphInputYielder fake_input(4, 1, -1, false, {"/CPU:0"}); in TEST_F() local
100 TrivialTestGraphInputYielder fake_input(4, 2, 1024 * 1024, false, in TEST_F() local
138 TrivialTestGraphInputYielder fake_input(4, 2, 1024 * 1024, false, {"/GPU:0"}); in TEST_F() local
Dgraph_properties_test.cc154 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
195 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
222 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
237 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
2071 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc135 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
154 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
175 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
198 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"/device:GPU:0"}); in TEST_F() local
222 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
239 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
730 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
753 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
775 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
796 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
[all …]
Dstatic_schedule_test.cc87 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
164 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
Dcommon_subgraph_elimination_test.cc57 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
Dimplementation_selector_test.cc46 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {CpuDevice}); in TEST_F() local
Dmodel_pruner_test.cc42 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
Dloop_optimizer_test.cc633 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
Ddependency_optimizer_test.cc55 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
Darithmetic_optimizer_test.cc101 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_test.cc30 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {{"CPU:0"}}); in TEST_F() local
Dgrappler_item_builder.cc175 Tensor fake_input(type, shape); in UpdatePlaceholderShape() local
375 Tensor fake_input(input.dtype(), shape); in GrapplerItemFromMetaGraphDef() local
Dgraph_view_test.cc128 TrivialTestGraphInputYielder fake_input(4, 2, 2, false, {"/CPU:0", "/GPU:0"}); in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc65 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
Dsingle_machine_test.cc73 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
112 TrivialTestGraphInputYielder fake_input(4, 1, 10, true, in TEST_F() local
123 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
/external/tensorflow/tensorflow/c/experimental/grappler/
Dgrappler_test.cc189 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST() local
237 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dscc_test.cc56 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.cc88 NodeDefBuilder& NodeDefBuilder::Input(FakeInputFunctor fake_input) { in Input()
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_analysis.cc620 Literal fake_input = Literal::CreateFromShape(while_cond_param->shape()); in ComputeWhileLoopTripCountUpperBound() local