Searched refs:layout_optimizer (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | memory_optimizer_test.py | 144 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
172 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
202 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
|
| D | layout_optimizer_test.py | 149 def _get_config(layout_optimizer=True): argument
150 if layout_optimizer:
152 layout_optimizer=rewriter_config_pb2.RewriterConfig.ON,
157 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
222 def _train(self, checkpoint_path, layout_optimizer=False, restore=False): argument
226 config=_get_config(layout_optimizer), graph=graph) as sess:
1486 layout_optimizer=rewriter_config_pb2.RewriterConfig.ON,
1504 layout_optimizer=rewriter_config_pb2.RewriterConfig.ON,
1527 checkpoint_path, restore=True, layout_optimizer=True)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | BUILD | 482 name = "layout_optimizer",
483 srcs = ["layout_optimizer.cc"],
485 "layout_optimizer.h",
507 ":layout_optimizer",
545 ":layout_optimizer",
|
| D | meta_optimizer.cc | 199 if (cfg_.layout_optimizer() != RewriterConfig::OFF) { in InitializeOptimizers()
684 rewrite_cfg.layout_optimizer() != RewriterConfig::OFF || in MetaOptimizerEnabled()
|
| D | layout_optimizer.cc | 2191 DataLayoutOptimizer layout_optimizer(graph_properties, *virtual_placer_, in Tune() local
2194 status = layout_optimizer.Optimize(); in Tune()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | conv2d_benchmark.py | 130 rewrite_options.layout_optimizer = (
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | rewriter_config.proto | 51 Toggle layout_optimizer = 1; field
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 93 rewrite_options.layout_optimizer = _rewriter_config_pb2.RewriterConfig.ON
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 88 rewriter_config.layout_optimizer = off
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | check_ops_test.py | 961 config.graph_options.rewrite_options.layout_optimizer = off
|