Home
last modified time | relevance | path

Searched defs:target_nodes (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlift_variables_test_pass.h71 const std::vector<std::string>& target_nodes, in Run()
81 const std::vector<std::string>& target_nodes, in Run()
91 const std::vector<std::string>& target_nodes, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dfake_session.cc113 const std::vector<std::string>& target_nodes, in Run()
124 const std::vector<std::string>& target_nodes, std::vector<Tensor>* outputs, in Run()
134 const std::vector<std::string>& target_nodes, std::vector<Tensor>* outputs, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/
H A Ddevice_tracer_test.cc185 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
209 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
238 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
287 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session_with_tracking_alloc_test.cc89 std::vector<string> target_nodes = {y_neg->name()}; in TEST() local
201 std::vector<string> target_nodes = {y_neg->name()}; in TestHWAccelerator() local
286 std::vector<string> target_nodes = {y_neg->name()}; in TEST() local
H A Ddirect_session.cc469 const std::vector<string>& target_nodes, in Run()
839 const std::vector<string>& target_nodes, in Run()
849 const std::vector<string>& target_nodes, in Run()
959 const std::vector<string>& target_nodes, in PRunSetup()
1483 gtl::ArraySlice<string> target_nodes, ExecutorsAndKeys** executors_and_keys, in GetOrCreateExecutors()
H A Dsession.cc48 const std::vector<string>& target_nodes, in PRunSetup()
H A Ddirect_session_test.cc143 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
214 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
246 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
719 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
785 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmemory_optimizer.cc166 std::unordered_set<NodeDef*> target_nodes; member
235 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in GetMaxDownstreamComponents()
303 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in AddRecomputeControlDependencyNodes()
397 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in RecomputeSubgraph()
H A Dscoped_allocator_optimizer_test.cc304 std::vector<string> target_nodes = {}; in ExecuteGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dtf_session_helper.cc63 const NameVector& target_nodes, in TF_Run_wrapper_helper()
167 const NameVector& target_nodes, TF_Status* out_status, in TF_Run_wrapper()
318 const NameVector& target_nodes, TF_Status* out_status, in TF_PRunSetup_wrapper()
H A Dsession_ref.cc258 const std::vector<string>& target_nodes, in RecordPRunSetup()
452 const std::vector<string>& target_nodes, in PRunSetup()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler_test.cc658 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
722 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.cc725 std::vector<std::string> target_nodes; member
854 const std::vector<std::string>& target_nodes) { in ImportSubgraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.h103 std::vector<std::string> target_nodes; member
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dgraph_util_impl.py207 def _bfs_for_reachable_nodes(target_nodes, name_to_input_name): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dsubgraph.cc192 const gtl::ArraySlice<string>& target_nodes) { in PruneForTargets()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_session.cc318 const std::vector<string>& target_nodes, in PRunSetup()
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A Dloader.cc381 const std::vector<string>& target_nodes, in PRunSetup()
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_io_utils.cc331 const std::vector<string>& target_nodes, in PublishDebugMetadata()