()
| 45 | # Disable rewrites to make sure we don't end up having to update this test |
| 46 | # whenever we implement new ones. |
| 47 | def NoRewriteSessionConfig(): |
| 48 | rewriter_config = rewriter_config_pb2.RewriterConfig( |
| 49 | disable_model_pruning=True, |
| 50 | arithmetic_optimization=rewriter_config_pb2.RewriterConfig.OFF, |
| 51 | dependency_optimization=rewriter_config_pb2.RewriterConfig.OFF, |
| 52 | function_optimization=rewriter_config_pb2.RewriterConfig.OFF) |
| 53 | graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config) |
| 54 | return config_pb2.ConfigProto(graph_options=graph_options) |
| 55 | |
| 56 | |
| 57 | def CompiledKernel(fn, *inputs, **kwargs): |
no outgoing calls
no test coverage detected