Searched refs:arg_source_tensors (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass_test.cc | 61 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()
|
| H A D | encapsulate_subgraphs_pass.h | 52 const std::vector<OutputTensor>& arg_source_tensors,
|
| H A D | encapsulate_subgraphs_pass.cc | 554 std::vector<OutputTensor> arg_source_tensors(args_by_src_.size()); in BuildFunctionDef() local
556 arg_source_tensors.at(arg.second) = arg.first; in BuildFunctionDef()
565 rewrite_subgraph_fn(arg_source_tensors, &graph_, &input_permutation, in BuildFunctionDef()
1199 [flr](const std::vector<OutputTensor>& arg_source_tensors, in Run()
|
| H A D | encapsulate_xla_computations_pass.cc | 114 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph() argument
|
| H A D | encapsulate_subgraphs_pass_test.cc | 795 [&guaranteed_consts](const std::vector<OutputTensor>& arg_source_tensors, in TEST()
840 [&guaranteed_consts](const std::vector<OutputTensor>& arg_source_tensors, in TEST()
|
| H A D | extract_outside_compilation_pass.cc | 2204 const std::vector<OutputTensor>& arg_source_tensors, in operator ()() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | encapsulate_tpu_computations_pass.cc | 77 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph() argument
87 TF_CHECK_OK(GetIndexAttr(n, arg_source_tensors.size(), &index)); in RewriteSubgraph()
89 arg_source_tensors.at(index).node->type_string() == kTPUReplicatedInput; in RewriteSubgraph()
91 if (!ret || !GetNodeAttr(arg_source_tensors.at(index).node->attrs(), in RewriteSubgraph()
|