This commit is contained in:
zong-shuai 2022-04-15 16:36:25 +08:00
parent 7541f981c1
commit 56d5e3e27d
1 changed files with 1 additions and 0 deletions

View File

@ -127,6 +127,7 @@ class UniformCandidateSamplerGpuKernelMod : public DeprecatedNativeGpuKernelMod
void ReleaseResource() override {
// Reset current seed.
set_input_.clear();
if (init_seed_ == 0 && cur_seed_ != 0) {
cur_seed_ = 0;
}