Searched refs:accum_val_persistent_ (Results 1 – 1 of 1) sorted by relevance
60 accum_val_persistent_ = new PersistentTensor(); in SparseConditionalAccumulator()66 if (accum_val_persistent_ != nullptr) delete accum_val_persistent_; in ~SparseConditionalAccumulator() local75 PersistentTensor* accum_val_persistent_ = nullptr; variable176 ctx->allocate_persistent(dtype_, grad_val->shape(), accum_val_persistent_, in AllocateAndAssignToAccumGradFunction()312 delete accum_val_persistent_; in AddToAccumGradFunction() local313 accum_val_persistent_ = tensor_sum_persistent; in AddToAccumGradFunction()