Searched refs:rewrite_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | regex_replace_op_test.py | 109 def testRegexReplaceDelegation(self, pattern_fn, rewrite_fn): argument
113 replace = rewrite_fn(".")
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 60 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
65 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
131 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
136 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
157 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
162 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
181 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
186 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
213 RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); in TEST() local
218 rewrite_fn(arg_source_tensors, &g, nullptr, nullptr, &call_node_def)); in TEST()
|
| D | extract_outside_compilation_pass.cc | 2364 auto rewrite_fn = absl::make_unique<RewriteOutsideCompilationSubgraphFn>( in ExtractOutsideCompilationForFunction() local
2368 outside_compilation_attr_name, *fbody->graph, *rewrite_fn, in ExtractOutsideCompilationForFunction()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 1017 def rewrite_fn(*args): function
1052 replicate_outputs = rewrite_fn()
1054 replicate_outputs = training_loop.repeat(iterations, rewrite_fn,
|