Home
last modified time | relevance | path

Searched refs:fake_input (Results 1 – 20 of 20) 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
48 CHECK(fake_input.NextItem(&item)); in TEST_F()
74 TrivialTestGraphInputYielder fake_input(4, 1, -1, false, {"/CPU:0"}); in TEST_F() local
76 CHECK(fake_input.NextItem(&item)); in TEST_F()
100 TrivialTestGraphInputYielder fake_input(4, 2, 1024 * 1024, false, in TEST_F() local
103 CHECK(fake_input.NextItem(&item)); in TEST_F()
138 TrivialTestGraphInputYielder fake_input(4, 2, 1024 * 1024, false, {"/GPU:0"}); in TEST_F() local
140 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dgraph_properties_test.cc123 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
126 CHECK(fake_input.NextItem(&item)); in TEST_F()
167 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
170 CHECK(fake_input.NextItem(&item)); in TEST_F()
194 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
197 CHECK(fake_input.NextItem(&item)); in TEST_F()
1659 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
1662 CHECK(fake_input.NextItem(&item)); 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() local
138 CHECK(fake_input.NextItem(&item)); in TEST_F()
155 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
157 CHECK(fake_input.NextItem(&item)); in TEST_F()
176 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
178 CHECK(fake_input.NextItem(&item)); in TEST_F()
199 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
201 CHECK(fake_input.NextItem(&item)); in TEST_F()
216 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
218 CHECK(fake_input.NextItem(&item)); in TEST_F()
[all …]
Dstatic_schedule_test.cc50 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
52 CHECK(fake_input.NextItem(&item)); in TEST_F()
126 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
128 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dimplementation_selector_test.cc44 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {CpuDevice}); in TEST_F() local
46 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dmodel_pruner_test.cc37 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
39 CHECK(fake_input.NextItem(&item)); in TEST_F()
Ddependency_optimizer_test.cc53 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
55 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dloop_optimizer_test.cc518 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
520 CHECK(fake_input.NextItem(&item)); in TEST_F()
Darithmetic_optimizer_test.cc92 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
94 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc274 Tensor fake_input(input.dtype(), shape); in GrapplerItemFromMetaGraphDef() local
275 InitializeTensor(input.dtype(), &fake_input); in GrapplerItemFromMetaGraphDef()
276 new_item->feed.emplace_back(NodeName(input.name()), fake_input); in GrapplerItemFromMetaGraphDef()
496 Tensor fake_input(type, shape); in GrapplerItemFromMetaGraphDef() local
497 InitializeTensor(type, &fake_input); in GrapplerItemFromMetaGraphDef()
502 new_item->feed.emplace_back(node.name(), fake_input); in GrapplerItemFromMetaGraphDef()
511 it->second = fake_input; in GrapplerItemFromMetaGraphDef()
Dgrappler_item_test.cc30 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {{"CPU:0"}}); in TEST_F() local
32 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dgraph_view_test.cc127 TrivialTestGraphInputYielder fake_input(4, 2, 2, false, {"/CPU:0", "/GPU:0"}); in TEST_F() local
129 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc64 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
67 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dsingle_machine_test.cc68 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
71 CHECK(fake_input.NextItem(&item)); in TEST_F()
101 TrivialTestGraphInputYielder fake_input(4, 1, 10, true, in TEST_F() local
104 CHECK(fake_input.NextItem(&item)); in TEST_F()
112 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
117 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_analysis.cc466 Literal fake_input = Literal::CreateFromShape(while_cond_param->shape()); in ComputeWhileLoopTripCountUpperBound() local
467 TF_CHECK_OK(fake_input.CopyFrom(while_body_indvar->literal(), in ComputeWhileLoopTripCountUpperBound()
471 evaluator.Evaluate(*while_cond, {std::move(fake_input)}); in ComputeWhileLoopTripCountUpperBound()
/external/tensorflow/tensorflow/core/grappler/utils/
Dscc_test.cc56 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
59 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.cc88 NodeDefBuilder& NodeDefBuilder::Input(FakeInputFunctor fake_input) { in Input() argument
90 Status status = fake_input(*op_def_, inputs_specified_, node_def_, this); in Input()
Dnode_def_builder.h85 NodeDefBuilder& Input(FakeInputFunctor fake_input);
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc678 Tensor fake_input(type, shape); in OptimizeGraph() local
679 item.feed.emplace_back(node.name(), fake_input); in OptimizeGraph()
/external/tensorflow/tensorflow/core/
DBUILD1588 "framework/fake_input.cc",
1596 "framework/fake_input.h",
1976 ":framework/fake_input.cc",
1977 ":framework/fake_input.h",
2011 "framework/fake_input.h",
2697 "framework/fake_input.*",
2797 "framework/fake_input.*",