Searched defs:rewrite_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| D | regex_replace_op_test.py | 105 def testRegexReplaceDelegation(self, pattern_fn, rewrite_fn): argument
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 60 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
131 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
157 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
181 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
213 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
|
| D | extract_outside_compilation_pass.cc | 2365 auto rewrite_fn = std::make_unique<RewriteOutsideCompilationSubgraphFn>( in ExtractOutsideCompilationForFunction() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 1075 def rewrite_fn(*args): function
|