Searched refs:lr_idx (Results 1 – 1 of 1) sorted by relevance
43 …onst std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx, int lr_idx, int grad_idx) in OptimizerKernel() argument44 : InnerKernel(parameter, inputs, outputs, ctx), lr_idx_(lr_idx), grad_idx_(grad_idx) {} in OptimizerKernel()