|
|
|
@@ -82,6 +82,9 @@ void SparseApplyAdamPSKernel::ReInit(const std::vector<AddressPtr> &inputs) { |
|
|
|
bool SparseApplyAdamPSKernel::Execute(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace, |
|
|
|
const std::vector<AddressPtr> &outputs) { |
|
|
|
ReInit(inputs); |
|
|
|
if (indices_size_ == 0) { |
|
|
|
return true; |
|
|
|
} |
|
|
|
return Launch(inputs, workspace, outputs); |
|
|
|
} |
|
|
|
|
|
|
|
|