/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_memory_test.cc | 46 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()
|
D | graph_properties_test.cc | 123 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/ |
D | meta_optimizer_test.cc | 136 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 …]
|
D | static_schedule_test.cc | 50 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()
|
D | implementation_selector_test.cc | 44 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {CpuDevice}); in TEST_F() local 46 CHECK(fake_input.NextItem(&item)); in TEST_F()
|
D | model_pruner_test.cc | 37 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local 39 CHECK(fake_input.NextItem(&item)); in TEST_F()
|
D | dependency_optimizer_test.cc | 53 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local 55 CHECK(fake_input.NextItem(&item)); in TEST_F()
|
D | loop_optimizer_test.cc | 518 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {"CPU:0"}); in TEST_F() local 520 CHECK(fake_input.NextItem(&item)); in TEST_F()
|
D | arithmetic_optimizer_test.cc | 92 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/ |
D | grappler_item_builder.cc | 274 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()
|
D | grappler_item_test.cc | 30 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {{"CPU:0"}}); in TEST_F() local 32 CHECK(fake_input.NextItem(&item)); in TEST_F()
|
D | graph_view_test.cc | 127 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/ |
D | virtual_cluster_test.cc | 64 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, in TEST_F() local 67 CHECK(fake_input.NextItem(&item)); in TEST_F()
|
D | single_machine_test.cc | 68 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/ |
D | while_loop_analysis.cc | 466 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/ |
D | scc_test.cc | 56 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/ |
D | node_def_builder.cc | 88 NodeDefBuilder& NodeDefBuilder::Input(FakeInputFunctor fake_input) { in Input() argument 90 Status status = fake_input(*op_def_, inputs_specified_, node_def_, this); in Input()
|
D | node_def_builder.h | 85 NodeDefBuilder& Input(FakeInputFunctor fake_input);
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | graph_execution_state.cc | 678 Tensor fake_input(type, shape); in OptimizeGraph() local 679 item.feed.emplace_back(node.name(), fake_input); in OptimizeGraph()
|
/external/tensorflow/tensorflow/core/ |
D | BUILD | 1588 "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.*",
|