Searched refs:layout_optimizer (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | memory_optimizer_test.py | 141 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
169 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
199 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
|
| D | layout_optimizer_test.py | 145 def _get_config(layout_optimizer=True): argument
146 if layout_optimizer:
148 layout_optimizer=rewriter_config_pb2.RewriterConfig.ON,
153 layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
239 def _train(self, checkpoint_path, layout_optimizer=False, restore=False): argument
243 config=_get_config(layout_optimizer), graph=graph) as sess:
2072 layout_optimizer=rewriter_config_pb2.RewriterConfig.ON,
2090 layout_optimizer=rewriter_config_pb2.RewriterConfig.ON,
2113 checkpoint_path, restore=True, layout_optimizer=True)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 226 /*optimization level*/ cfg_.layout_optimizer(), in MakeNewOptimizer()
415 if (BOTH_NOT_OFF(layout_optimizer)) { in InitializeOptimizers()
416 if (USER_IS_EXPERIMENTAL_MLIR(layout_optimizer) || in InitializeOptimizers()
417 USER_IS_EXPERIMENTAL_BOTH(layout_optimizer)) { in InitializeOptimizers()
421 /*optimization level*/ cfg_.layout_optimizer(), in InitializeOptimizers()
628 PRINT_CFG(layout_optimizer) in PrintUserAndPluginConfigs()
1332 rewrite_cfg.layout_optimizer() != RewriterConfig::OFF || in MetaOptimizerEnabled()
|
| /external/tensorflow/tensorflow/c/experimental/grappler/ |
| D | grappler.h | 114 TF_TriState layout_optimizer; member
|
| D | grappler.cc | 130 CONFIG_TOGGLE(layout_optimizer); in CGraphOptimizerRegister()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | conv2d_benchmark.py | 126 rewrite_options.layout_optimizer = (
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | utils.py | 43 rewriter_config.layout_optimizer = off
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | rewriter_config.proto | 69 Toggle layout_optimizer = 1; field
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | check_ops_test.py | 1024 config.graph_options.rewrite_options.layout_optimizer = off
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5863 tensorflow::grappler::GenericLayoutOptimizer layout_optimizer("NCHW"); in ConvertGraphDefToEngine() local
5865 layout_optimizer.Optimize(cluster, grappler_item, &graph)); in ConvertGraphDefToEngine()
|