Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
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
Dencapsulate_xla_computations_pass.cc109 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph()
Dencapsulate_subgraphs_pass.cc578 std::vector<OutputTensor> arg_source_tensors(args_by_src_.size()); in BuildFunctionDef() local
1216 std::vector<int>* output_permutation, NodeDef* node) { in Run()
Dencapsulate_subgraphs_pass_test.cc799 NodeDef* call_def) { in TEST()
844 NodeDef* call_def) { in TEST()
Dextract_outside_compilation_pass.cc2203 const std::vector<OutputTensor>& arg_source_tensors, in operator ()()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc76 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph()