Home
last modified time | relevance | path

Searched refs:fake_input (Results 1 – 25 of 25) 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()
237 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
240 CHECK(fake_input.NextItem(&item)); in TEST_F()
2071 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
2074 CHECK(fake_input.NextItem(&item)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc135 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
137 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
154 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
156 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
175 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
177 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
198 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"/device:GPU:0"}); in TEST_F() local
200 ASSERT_TRUE(fake_input.NextItem(&item)); in TEST_F()
222 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F() local
224 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.cc101 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local
103 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()
375 Tensor fake_input(input.dtype(), shape); in GrapplerItemFromMetaGraphDef() local
376 InitializeTensor(input.dtype(), &fake_input); in GrapplerItemFromMetaGraphDef()
377 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.cc73 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local
76 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/c/experimental/grappler/
Dgrappler_test.cc189 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST() local
192 CHECK(fake_input.NextItem(&item)); in TEST()
237 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST() local
240 CHECK(fake_input.NextItem(&item)); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc708 Tensor fake_input(DT_INVALID, {0}); in OptimizeGraph()
709 item.feed.emplace_back(feed.ToString(), fake_input); in OptimizeGraph()
734 Tensor fake_input(DT_INVALID, {0}); in OptimizeGraph()
735 item.feed.emplace_back(node->name(), fake_input); in OptimizeGraph()
761 Tensor fake_input(type, shape); in OptimizeGraph()
762 item.feed.emplace_back(node->name(), fake_input); in OptimizeGraph()
/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/compiler/xla/service/
Dwhile_loop_analysis.cc620 Literal fake_input = Literal::CreateFromShape(while_cond_param->shape()); in ComputeWhileLoopTripCountUpperBound() local
621 TF_CHECK_OK(fake_input.CopyFrom(while_body_indvar->literal(), in ComputeWhileLoopTripCountUpperBound()
625 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()
Dnode_def_builder.h87 NodeDefBuilder& Input(FakeInputFunctor fake_input);
DBUILD454 "fake_input.h",
463 "fake_input.cc",
613 name = "fake_input",
615 srcs = ["fake_input.cc"],
616 hdrs = ["fake_input.h"],
1123 "fake_input.h",
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
DBUILD333 "//tensorflow/core/framework:fake_input",
375 "//tensorflow/core/framework:fake_input",
/external/tensorflow/tensorflow/core/kernels/mkl/
DBUILD135 "//tensorflow/core/framework:fake_input",
/external/tensorflow/tensorflow/core/
DBUILD835 "//tensorflow/core/framework:fake_input.h",
855 "//tensorflow/core/framework:fake_input",