Home
last modified time | relevance | path

Searched refs:memory_optimization (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py60 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))
88 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL,
148 memory_optimization=(
176 memory_optimization=rewriter_config_pb2.RewriterConfig
205 memory_optimization=rewriter_config_pb2.RewriterConfig
226 memory_optimization=rewriter_config_pb2.RewriterConfig.HEURISTICS)
298 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL)
319 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dutils.py44 rewriter_config.memory_optimization = (
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc341 if (MemoryOptimizerEnabled(cfg_.memory_optimization(), in InitializeOptimizers()
343 PLUGIN_NOT_OFF(memory_optimization)) { in InitializeOptimizers()
347 MakeUnique<MemoryOptimizer>(cfg_.memory_optimization())); in InitializeOptimizers()
350 cfg_.memory_optimization(), in InitializeOptimizers()
516 MemoryOptimizerEnabled(cfg_.memory_optimization(), in PrintUserAndPluginConfigs()
1155 rewrite_cfg.memory_optimization() != RewriterConfig::NO_MEM_OPT || in MetaOptimizerEnabled()
/external/tensorflow/tensorflow/c/experimental/grappler/
Dgrappler.h118 TF_TriState memory_optimization; member
Dgrappler.cc133 CONFIG_TOGGLE(memory_optimization); in CGraphOptimizerRegister()
/external/tensorflow/tensorflow/core/protobuf/
Drewriter_config.proto162 MemOptType memory_optimization = 4; field
166 // heuristics depending on memory_optimization), but the nodes themselves will
/external/tensorflow/tensorflow/python/kernel_tests/
Dwhile_v2_test.py2043 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))