Searched refs:vspace_ (Results 1 – 1 of 1) sorted by relevance
235 : vspace_(vspace), use_batch_(use_batch) { in ForwardAccumulator()241 vspace_.DeleteGradient(accumulated.second); in ~ForwardAccumulator()340 const VSpace<Gradient, BackwardFunction, TapeTensor>& vspace_; variable959 TF_RETURN_IF_ERROR(vspace_.BuildOnesLike(output_tensor, &aid)); in ForwardpropFromTape()967 int64 aid_id = vspace_.TensorId(aid); in ForwardpropFromTape()975 this->vspace_.DeleteGradient(tensor); in ForwardpropFromTape()983 TF_RETURN_IF_ERROR(vspace_.CallBackwardFunction( in ForwardpropFromTape()1001 int64 tensor_id = vspace_.TensorId(grad_tensor); in ForwardpropFromTape()1005 tensor_id, vspace_.TapeTensorFromGradient(grad_tensor)); in ForwardpropFromTape()1010 vspace_.MarkAsResult(in_grad); in ForwardpropFromTape()[all …]