Searched refs:rewrite_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | quantize_graph_test.py | 95 def _TestRewrite(self, rewrite_fn): argument
103 rewrite_fn(graph)
112 def _TestDefaultGraph(self, rewrite_fn): argument
119 rewrite_fn()
128 def _TestWithPostActivationBypass(self, rewrite_fn): argument
133 rewrite_fn()
142 def _TestQuantDelay(self, rewrite_fn): argument
146 rewrite_fn(quant_delay=quant_delay)
160 def _TestTrainingOpsCheck(self, rewrite_fn): argument
168 rewrite_fn()
[all …]
|
| /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 | 1629 RewriteOutsideCompilationSubgraphFn rewrite_fn( in ExtractOutsideCompilationForFunction() local
1632 outside_compilation_attr_name, "", *fbody->graph, rewrite_fn, in ExtractOutsideCompilationForFunction()
|
| /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/python/distribute/ |
| D | tpu_strategy.py | 319 def rewrite_fn(*args): function
349 replicate_outputs = rewrite_fn()
351 replicate_outputs = training_loop.repeat(iterations, rewrite_fn,
|