Searched refs:graph_runner (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_runner_test.cc | 49 GraphRunner graph_runner(Env::Default()); in TEST() local
51 Status s = graph_runner.Run(root.graph(), nullptr, {}, {c.name()}, &outputs); in TEST()
75 GraphRunner graph_runner(Env::Default()); in TEST() local
77 graph_runner.Run(root.graph(), nullptr, inputs, {"add:0"}, &outputs); in TEST()
87 GraphRunner graph_runner(Env::Default()); in TEST() local
89 Status s = graph_runner.Run(root.graph(), nullptr, {}, {c.name(), pi.name()}, in TEST()
109 GraphRunner graph_runner(Env::Default()); in TEST() local
112 graph_runner.Run(root.graph(), nullptr, inputs, {"add:0"}, &outputs); in TEST()
|
| D | eval_const_tensor.cc | 311 Tensor* result, GraphRunner* graph_runner, in EvaluateConstantTensor() argument
348 if (graph_runner == nullptr) { in EvaluateConstantTensor()
351 graph_runner = graph_runner_storage.get(); in EvaluateConstantTensor()
356 Status s = graph_runner->Run(&subgraph, nullptr /* function_library */, in EvaluateConstantTensor()
|
| D | eval_const_tensor.h | 59 Tensor* result, GraphRunner* graph_runner = nullptr,
|
| D | constant_folding.cc | 624 auto graph_runner = std::unique_ptr<GraphRunner>(new GraphRunner(env)); in ConstantFold() local
627 auto delete_tensors = gtl::MakeCleanup([&graph_runner, &outputs] { in ConstantFold()
630 graph_runner.reset(nullptr); in ConstantFold()
634 graph_runner->Run(constant_graph.get(), function_library, {} /* inputs*/, in ConstantFold()
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 67 GraphRunner graph_runner(device); in FromGraph() local
68 TF_RETURN_IF_ERROR(graph_runner.Run(&graph, pflr->GetFLR("/device:CPU:0"), in FromGraph()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | graph_rewrite_dataset.cc | 69 GraphRunner graph_runner(ctx->function_library()->device()); in Optimize() local
72 graph_runner.Run(&graph, lib_, input_list, {output_node}, &outputs)); in Optimize()
|
| D | iterator_ops.cc | 140 GraphRunner graph_runner(ctx->env()); in Restore() local
163 graph_runner.Run(&graph, new_state->lib, {}, {output_node}, &outputs)); in Restore()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 2982 "common_runtime/graph_runner.h",
3107 "common_runtime/graph_runner.cc",
|