Home
last modified time | relevance | path

Searched refs:target_nodes (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables_test_pass.cc68 const std::vector<std::string>& target_nodes, in Run() argument
71 return Run(tensorflow::RunOptions(), inputs, output_names, target_nodes, in Run()
78 const std::vector<std::string>& target_nodes, in Run() argument
81 return Run(run_options, inputs, output_names, target_nodes, outputs, in Run()
88 const std::vector<std::string>& target_nodes, in Run() argument
Dlift_variables_test_pass.h71 const std::vector<std::string>& target_nodes, in Run() argument
74 return Run(tensorflow::RunOptions(), inputs, output_names, target_nodes, in Run()
81 const std::vector<std::string>& target_nodes, in Run() argument
84 return Run(run_options, inputs, output_names, target_nodes, outputs, in Run()
91 const std::vector<std::string>& target_nodes, in Run() argument
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc178 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
182 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
205 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
209 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
230 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
237 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST_F()
258 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
262 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc62 const NameVector& target_nodes, in TF_Run_wrapper_helper() argument
114 const_cast<const char**>(target_nodes.data()), target_nodes.size(), in TF_Run_wrapper_helper()
120 const_cast<const char**>(target_nodes.data()), target_nodes.size(), in TF_Run_wrapper_helper()
166 const NameVector& target_nodes, TF_Status* out_status, in TF_Run_wrapper() argument
169 target_nodes, out_status, out_values, run_outputs); in TF_Run_wrapper()
317 const NameVector& target_nodes, TF_Status* out_status, in TF_PRunSetup_wrapper() argument
323 const_cast<const char**>(target_nodes.data()), target_nodes.size(), in TF_PRunSetup_wrapper()
Dsession_ref.cc257 const std::vector<string>& target_nodes, in RecordPRunSetup() argument
268 for (auto& target : target_nodes) { in RecordPRunSetup()
271 RUN_WITH_TIMESTAMP(PRunSetup, input_names, output_names, target_nodes, in RecordPRunSetup()
451 const std::vector<string>& target_nodes, in PRunSetup() argument
453 LOG_AND_RUN_OPERATION(PRunSetup, input_names, output_names, target_nodes, in PRunSetup()
Dtf_session_helper.h62 const NameVector& target_nodes, TF_Status* out_status,
99 const NameVector& target_nodes, TF_Status* out_status,
Dsession_ref.h59 const std::vector<string>& target_nodes,
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc89 std::vector<string> target_nodes = {y_neg->name()}; in TEST() local
92 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST()
200 std::vector<string> target_nodes = {y_neg->name()}; in TestHWAccelerator() local
209 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TestHWAccelerator()
284 std::vector<string> target_nodes = {y_neg->name()}; in TEST() local
288 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST()
Ddirect_session.h76 const std::vector<string>& target_nodes,
83 const std::vector<string>& target_nodes,
91 const std::vector<string>& target_nodes, std::vector<Tensor>* outputs,
99 const std::vector<string>& target_nodes,
245 gtl::ArraySlice<string> target_nodes,
Ddebugger_state_interface.h55 const std::vector<string>& target_nodes) = 0;
Ddirect_session.cc458 const std::vector<string>& target_nodes, in Run() argument
461 return Run(RunOptions(), inputs, output_names, target_nodes, outputs, in Run()
822 const std::vector<string>& target_nodes, in Run() argument
825 return Run(run_options, inputs, output_names, target_nodes, outputs, in Run()
832 const std::vector<string>& target_nodes, in Run() argument
857 target_nodes, &executors_and_keys, in Run()
942 const std::vector<string>& target_nodes, in PRunSetup() argument
957 target_nodes, &executors_and_keys, in PRunSetup()
1466 gtl::ArraySlice<string> target_nodes, ExecutorsAndKeys** executors_and_keys, in GetOrCreateExecutors() argument
1482 absl::StrJoin(target_nodes, ","), "/", run_state_args->is_partial_run, in GetOrCreateExecutors()
[all …]
Dsession.cc48 const std::vector<string>& target_nodes, in PRunSetup() argument
Ddirect_session_test.cc142 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
144 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
213 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
215 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
245 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
247 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
261 s = session->Run(run_options, inputs, output_names, target_nodes, &outputs, in TEST_F()
718 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
727 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST_F()
784 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc166 std::unordered_set<NodeDef*> target_nodes; member
199 current_recomputation.target_nodes.insert(output); in GetOpGroupsToRecompute()
220 if (current_recomputation.target_nodes.empty()) { in GetOpGroupsToRecompute()
235 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in GetMaxDownstreamComponents() argument
243 if (target_nodes.count(output) != 0) { in GetMaxDownstreamComponents()
303 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in AddRecomputeControlDependencyNodes() argument
330 for (const NodeDef* target_node : target_nodes) { in AddRecomputeControlDependencyNodes()
397 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in RecomputeSubgraph() argument
404 GetMaxDownstreamComponents(recomputed_source_nodes, target_nodes, in RecomputeSubgraph()
411 AddRecomputeControlDependencyNodes(recomputed_source_nodes, target_nodes, in RecomputeSubgraph()
[all …]
Dscoped_allocator_optimizer_test.cc304 std::vector<string> target_nodes = {}; in ExecuteGraph() local
305 Status s = session->Run(inputs, output_names, target_nodes, outputs); in ExecuteGraph()
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py167 def _bfs_for_reachable_nodes(target_nodes, name_to_input_name): argument
171 next_to_visit = target_nodes[:]
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc658 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
665 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST_F()
722 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
732 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dsubgraph.cc192 const gtl::ArraySlice<string>& target_nodes) { in PruneForTargets() argument
200 for (const string& s : target_nodes) { in PruneForTargets()
/external/tensorflow/tensorflow/core/debug/
Ddebug_io_utils.cc331 const std::vector<string>& target_nodes, in PublishDebugMetadata() argument
357 for (size_t i = 0; i < target_nodes.size(); ++i) { in PublishDebugMetadata()
358 oss << "\"" << target_nodes[i] << "\""; in PublishDebugMetadata()
359 if (i < target_nodes.size() - 1) { in PublishDebugMetadata()
Ddebug_io_utils.h66 const std::vector<string>& target_nodes,
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.h91 const std::vector<string>& target_nodes,
Dgrpc_session.cc317 const std::vector<string>& target_nodes, in PRunSetup() argument
330 for (const string& target : target_nodes) { in PRunSetup()
/external/tensorflow/tensorflow/core/public/
Dsession.h197 const std::vector<std::string>& target_nodes,
/external/llvm-project/clang/utils/analyzer/
Dexploded-graph-rewriter.py935 def __init__(self, target_nodes): argument
936 self._target_nodes = target_nodes
951 def parse_target_nodes(target_nodes, graph): argument
953 for node in target_nodes.split(',')]
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py280 self.assertEqual([], results.dump.core_metadata.target_nodes)
1147 self.assertEqual([], dump1.core_metadata.target_nodes)
1162 self.assertEqual([], dump2.core_metadata.target_nodes)
1179 self.assertEqual([], dump3.core_metadata.target_nodes)
1210 self.assertEqual([], dump.core_metadata.target_nodes)

12