Home
last modified time | relevance | path

Searched refs:arg_source_tensors (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc61 std::vector<OutputTensor> arg_source_tensors; in TEST() local
65 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
132 std::vector<OutputTensor> arg_source_tensors; in TEST() local
136 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
158 std::vector<OutputTensor> arg_source_tensors; in TEST() local
162 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
182 std::vector<OutputTensor> arg_source_tensors; in TEST() local
186 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
214 std::vector<OutputTensor> arg_source_tensors; in TEST() local
218 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
Dencapsulate_subgraphs_pass.h52 const std::vector<OutputTensor>& arg_source_tensors,
Dencapsulate_subgraphs_pass.cc578 std::vector<OutputTensor> arg_source_tensors(args_by_src_.size()); in BuildFunctionDef() local
580 arg_source_tensors.at(arg.second) = arg.first; in BuildFunctionDef()
589 rewrite_subgraph_fn(arg_source_tensors, &graph_, &input_permutation, in BuildFunctionDef()
1213 [flr](const std::vector<OutputTensor>& arg_source_tensors, in Run()
Dencapsulate_xla_computations_pass.cc109 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph() argument
Dencapsulate_subgraphs_pass_test.cc795 [&guaranteed_consts](const std::vector<OutputTensor>& arg_source_tensors, in TEST()
840 [&guaranteed_consts](const std::vector<OutputTensor>& arg_source_tensors, in TEST()
Dextract_outside_compilation_pass.cc2203 const std::vector<OutputTensor>& arg_source_tensors, in operator ()() argument
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc76 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph() argument
86 TF_CHECK_OK(GetIndexAttr(n, arg_source_tensors.size(), &index)); in RewriteSubgraph()
88 arg_source_tensors.at(index).node->type_string() == kTPUReplicatedInput; in RewriteSubgraph()
90 if (!ret || !GetNodeAttr(arg_source_tensors.at(index).node->attrs(), in RewriteSubgraph()
102 TF_CHECK_OK(GetIndexAttr(n, arg_source_tensors.size(), &index)); in RewriteSubgraph()
103 if (arg_source_tensors.at(index).node->type_string() != in RewriteSubgraph()
108 TF_CHECK_OK(GetNodeAttr(arg_source_tensors.at(index).node->attrs(), "index", in RewriteSubgraph()