Home
last modified time | relevance | path

Searched defs:arg_source_tensors (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass_test.cc61 std::vector<OutputTensor> arg_source_tensors; in TEST() local
132 std::vector<OutputTensor> arg_source_tensors; in TEST() local
158 std::vector<OutputTensor> arg_source_tensors; in TEST() local
182 std::vector<OutputTensor> arg_source_tensors; in TEST() local
214 std::vector<OutputTensor> arg_source_tensors; in TEST() local
H A Dencapsulate_subgraphs_pass.cc554 std::vector<OutputTensor> arg_source_tensors(args_by_src_.size()); in BuildFunctionDef() local
1202 std::vector<int>* output_permutation, NodeDef* node) { in Run()
H A Dencapsulate_xla_computations_pass.cc114 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph()
H A Dencapsulate_subgraphs_pass_test.cc799 NodeDef* call_def) { in TEST()
844 NodeDef* call_def) { in TEST()
H A Dextract_outside_compilation_pass.cc2204 const std::vector<OutputTensor>& arg_source_tensors, in operator ()()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc77 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph()