Searched refs:rewrite_fn (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A 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()
|
| H A D | extract_outside_compilation_pass.cc | 2365 auto rewrite_fn = std::make_unique<RewriteOutsideCompilationSubgraphFn>( in ExtractOutsideCompilationForFunction() local
2369 outside_compilation_attr_name, *fbody->graph, *rewrite_fn, in ExtractOutsideCompilationForFunction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| H A D | regex_replace_op_test.py | 105 def testRegexReplaceDelegation(self, pattern_fn, rewrite_fn): argument
109 replace = rewrite_fn(".")
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | tpu_strategy.py | 1075 def rewrite_fn(*args): function
1110 replicate_outputs = rewrite_fn()
1112 replicate_outputs = training_loop.repeat(iterations, rewrite_fn,
|