Searched refs:RewriteOutsideCompilationSubgraphFn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.h | 44 class RewriteOutsideCompilationSubgraphFn {
46 RewriteOutsideCompilationSubgraphFn( in RewriteOutsideCompilationSubgraphFn() function
|
| D | extract_outside_compilation_pass_test.cc | 60 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST()
131 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST()
157 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST()
181 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST()
213 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST()
|
| D | extract_outside_compilation_pass.cc | 2202 Status RewriteOutsideCompilationSubgraphFn::operator()( in operator ()()
2364 auto rewrite_fn = absl::make_unique<RewriteOutsideCompilationSubgraphFn>( in ExtractOutsideCompilationForFunction()
|