// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.//// Licensed under the Apache License, Version 2.0 (the "License");// you may not use this file except in compliance with the License.// You may obtain a copy of the License at//// http://www.apache.org/licenses/LICENSE-2.0//// Unless required by applicable law or agreed to in writing, software// distributed under the License is distributed on an "AS IS" BASIS,// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.// See the License for the specific language governing permissions and// limitations under the License.#pragma once#include "paddle/phi/core/dense_tensor.h"namespacephi{template<typenameT,typenameContext>voidAdamaxKernel(constContext&dev_ctx,constDenseTensor¶m,constDenseTensor&grad,constDenseTensor&learning_rate,constDenseTensor&moment,constDenseTensor&inf_norm,constDenseTensor&beta1_pow,floatbeta1,floatbeta2,floatepsilon,DenseTensor*param_out,DenseTensor*moment_out,DenseTensor*inf_norm_out);}// namespace phi