Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
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()
Dlift_variables_test_pass.cc68 const std::vector<std::string>& target_nodes, in Run()
78 const std::vector<std::string>& target_nodes, in Run()
88 const std::vector<std::string>& target_nodes, in Run()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc178 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
205 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
230 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
258 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc89 std::vector<string> target_nodes = {y_neg->name()}; in TEST() local
200 std::vector<string> target_nodes = {y_neg->name()}; in TestHWAccelerator() local
284 std::vector<string> target_nodes = {y_neg->name()}; in TEST() local
Dsession.cc48 const std::vector<string>& target_nodes, in PRunSetup()
Ddirect_session.cc458 const std::vector<string>& target_nodes, in Run()
822 const std::vector<string>& target_nodes, in Run()
832 const std::vector<string>& target_nodes, in Run()
942 const std::vector<string>& target_nodes, in PRunSetup()
1466 gtl::ArraySlice<string> target_nodes, ExecutorsAndKeys** executors_and_keys, in GetOrCreateExecutors()
Ddirect_session_test.cc142 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
213 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
245 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
718 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
784 std::vector<string> target_nodes = {y_neg_}; in TEST_F() local
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py167 def _bfs_for_reachable_nodes(target_nodes, name_to_input_name): argument
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc62 const NameVector& target_nodes, in TF_Run_wrapper_helper()
166 const NameVector& target_nodes, TF_Status* out_status, in TF_Run_wrapper()
317 const NameVector& target_nodes, TF_Status* out_status, in TF_PRunSetup_wrapper()
Dsession_ref.cc257 const std::vector<string>& target_nodes, in RecordPRunSetup()
451 const std::vector<string>& target_nodes, in PRunSetup()
/external/tensorflow/tensorflow/core/grappler/optimizers/
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()
Dscoped_allocator_optimizer_test.cc304 std::vector<string> target_nodes = {}; in ExecuteGraph() local
/external/tensorflow/tensorflow/core/graph/
Dsubgraph.cc192 const gtl::ArraySlice<string>& target_nodes) { in PruneForTargets()
/external/tensorflow/tensorflow/core/framework/
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
/external/llvm-project/clang/utils/analyzer/
Dexploded-graph-rewriter.py935 def __init__(self, target_nodes): argument
951 def parse_target_nodes(target_nodes, graph): argument
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc353 const std::vector<string>& target_nodes, in PRunSetup()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc317 const std::vector<string>& target_nodes, in PRunSetup()
/external/tensorflow/tensorflow/core/debug/
Ddebug_io_utils.cc331 const std::vector<string>& target_nodes, in PublishDebugMetadata()