Home
last modified time | relevance | path

Searched refs:fake_input (Results 1 – 23 of 23) 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.cc154 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
157 CHECK(fake_input.NextItem(&item)); in TEST_F()
195 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
198 CHECK(fake_input.NextItem(&item)); in TEST_F()
222 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
225 CHECK(fake_input.NextItem(&item)); in TEST_F()
1901 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
1904 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc141 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
143 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
160 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
162 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
181 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
183 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
204 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
206 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
221 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
223 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
[all …]
Dstatic_schedule_test.cc87 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
89 CHECK(fake_input.NextItem(&item)); in TEST_F()
164 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
166 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dcommon_subgraph_elimination_test.cc57 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
59 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dimplementation_selector_test.cc46 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {CpuDevice}); in TEST_F() local
48 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dmodel_pruner_test.cc42 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
44 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
Ddependency_optimizer_test.cc55 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
57 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dloop_optimizer_test.cc633 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
635 CHECK(fake_input.NextItem(&item)); in TEST_F()
Darithmetic_optimizer_test.cc98 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
100 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/
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()
Dgrappler_item_builder.cc175 Tensor fake_input(type, shape); in UpdatePlaceholderShape() local
176 InitializeTensor(type, &fake_input); in UpdatePlaceholderShape()
181 new_item->feed.emplace_back(node->name(), fake_input); in UpdatePlaceholderShape()
190 it->second = fake_input; in UpdatePlaceholderShape()
374 Tensor fake_input(input.dtype(), shape); in GrapplerItemFromMetaGraphDef() local
375 InitializeTensor(input.dtype(), &fake_input); in GrapplerItemFromMetaGraphDef()
376 new_item->feed.emplace_back(NodeName(input.name()), fake_input); in GrapplerItemFromMetaGraphDef()
Dgraph_view_test.cc128 TrivialTestGraphInputYielder fake_input(4, 2, 2, false, {"/CPU:0", "/GPU:0"}); in TEST_F() local
130 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc65 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
68 CHECK(fake_input.NextItem(&item)); in TEST_F()
Dsingle_machine_test.cc74 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
77 CHECK(fake_input.NextItem(&item)); in TEST_F()
112 TrivialTestGraphInputYielder fake_input(4, 1, 10, true, in TEST_F() local
115 CHECK(fake_input.NextItem(&item)); in TEST_F()
123 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
128 CHECK(fake_input.NextItem(&item)); in TEST_F()
/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/common_runtime/
Dgraph_execution_state.cc703 Tensor fake_input(DT_INVALID, {0}); in OptimizeGraph()
704 item.feed.emplace_back(feed.ToString(), fake_input); in OptimizeGraph()
729 Tensor fake_input(DT_INVALID, {0}); in OptimizeGraph()
730 item.feed.emplace_back(node->name(), fake_input); in OptimizeGraph()
756 Tensor fake_input(type, shape); in OptimizeGraph()
757 item.feed.emplace_back(node->name(), fake_input); in OptimizeGraph()
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_analysis.cc608 Literal fake_input = Literal::CreateFromShape(while_cond_param->shape()); in ComputeWhileLoopTripCountUpperBound() local
609 TF_CHECK_OK(fake_input.CopyFrom(while_body_indvar->literal(), in ComputeWhileLoopTripCountUpperBound()
613 evaluator.Evaluate(*while_cond, {std::move(fake_input)}); in ComputeWhileLoopTripCountUpperBound()
/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()
DBUILD440 "fake_input.h",
449 "fake_input.cc",
597 name = "fake_input",
599 srcs = ["fake_input.cc"],
600 hdrs = ["fake_input.h"],
1042 "fake_input.h",
Dnode_def_builder.h87 NodeDefBuilder& Input(FakeInputFunctor fake_input);
/external/tensorflow/tensorflow/core/kernels/mkl/
DBUILD120 "//tensorflow/core/framework:fake_input",
/external/tensorflow/tensorflow/core/
DBUILD791 "//tensorflow/core/framework:fake_input.h",
811 "//tensorflow/core/framework:fake_input",