Searched refs:rewrite_subgraph_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.h | 77 const RewriteSubgraphFn& rewrite_subgraph_fn, bool reuse_existing_functions,
|
| D | encapsulate_subgraphs_pass.cc | 163 Status BuildFunctionDefs(const RewriteSubgraphFn& rewrite_subgraph_fn,
204 const RewriteSubgraphFn& rewrite_subgraph_fn,
568 const string& name_in, const RewriteSubgraphFn& rewrite_subgraph_fn, in BuildFunctionDef() argument
577 if (rewrite_subgraph_fn) { in BuildFunctionDef()
589 rewrite_subgraph_fn(arg_source_tensors, &graph_, &input_permutation, in BuildFunctionDef()
811 const RewriteSubgraphFn& rewrite_subgraph_fn, bool reuse_existing_functions, in BuildFunctionDefs() argument
817 name, rewrite_subgraph_fn, reuse_existing_functions, library)); in BuildFunctionDefs()
1112 const RewriteSubgraphFn& rewrite_subgraph_fn, bool reuse_existing_functions, in EncapsulateSubgraphsInFunctions() argument
1119 rewrite_subgraph_fn, reuse_existing_functions, library)); in EncapsulateSubgraphsInFunctions()
|