(model, engine="", **kwargs)
| 2243 | |
| 2244 | |
| 2245 | def build_gftrl(model, engine="", **kwargs): |
| 2246 | if engine == "SIMD": |
| 2247 | assert core.IsOperator("GFtrl_ENGINE_SIMD") |
| 2248 | gftrl_optimizer = GFtrlOptimizer(engine=engine, **kwargs) |
| 2249 | return _build(model, gftrl_optimizer) |
| 2250 | |
| 2251 | |
| 2252 | def build_adagrad( |
searching dependent graphs…