| 224 | */ |
| 225 | template<class Float, OptimizerType optimizer_type> |
| 226 | __host__ __device__ decltype(&Optimizer::sgd_update<Float>) get_update_function() { |
| 227 | switch (optimizer_type) { |
| 228 | case kSGD: |
| 229 | return &Optimizer::sgd_update<Float>; |
| 230 | default: |
| 231 | return nullptr; |
| 232 | } |
| 233 | } |
| 234 | |
| 235 | /** |
| 236 | * @brief Compile-time binding of 1-moment optimizers |
nothing calls this directly
no outgoing calls
no test coverage detected