| 1065 | |
| 1066 | template <typename Gradient, typename BackwardFunction, typename TapeTensor> |
| 1067 | void ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::DeleteGradient( |
| 1068 | int64 tensor_id) { |
| 1069 | auto existing = accumulated_gradients_.find(tensor_id); |
| 1070 | if (existing != accumulated_gradients_.end()) { |
| 1071 | vspace_.DeleteGradient(existing->second); |
| 1072 | accumulated_gradients_.erase(existing); |
| 1073 | } |
| 1074 | } |
| 1075 | |
| 1076 | template <typename Gradient, typename BackwardFunction, typename TapeTensor> |
| 1077 | Gradient* ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::FetchJVP( |
no test coverage detected