Searched refs:MK_OPT (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 120 #define MK_OPT(NAME, VALUE) \ macro
125 MK_OPT("pruning", new ModelPruner()); in MakeNewOptimizer()
126 MK_OPT("function", new FunctionOptimizer(cfg_.function_optimization())); in MakeNewOptimizer()
127 MK_OPT("constfold", new ConstantFolding(cpu_device_)); in MakeNewOptimizer()
128 MK_OPT("shape", new ShapeOptimizer()); in MakeNewOptimizer()
129 MK_OPT("remap", new Remapper(cfg_.remapping())); in MakeNewOptimizer()
130 MK_OPT("layout", new LayoutOptimizer()); in MakeNewOptimizer()
131 MK_OPT("memory", new MemoryOptimizer(RewriterConfig::MANUAL)); in MakeNewOptimizer()
132 MK_OPT("arithmetic", new ArithmeticOptimizer(cfg_.arithmetic_optimization())); in MakeNewOptimizer()
133 MK_OPT("autoparallel", new AutoParallel(cfg_.auto_parallel().num_replicas())); in MakeNewOptimizer()
[all …]
|