Searched refs:memory_optimization (Results 1 – 7 of 7) sorted by relevance
60 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))88 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL,148 memory_optimization=(176 memory_optimization=rewriter_config_pb2.RewriterConfig205 memory_optimization=rewriter_config_pb2.RewriterConfig226 memory_optimization=rewriter_config_pb2.RewriterConfig.HEURISTICS)298 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL)319 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))
44 rewriter_config.memory_optimization = (
341 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()
118 TF_TriState memory_optimization; member
133 CONFIG_TOGGLE(memory_optimization); in CGraphOptimizerRegister()
162 MemOptType memory_optimization = 4; field166 // heuristics depending on memory_optimization), but the nodes themselves will
2043 memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL))