Searched refs:forward_grads (Results 1 – 1 of 1) sorted by relevance
1084 std::vector<Gradient*> forward_grads; in Accumulate() local1088 forward_grads.resize(output_tensors.size()); in Accumulate()1091 backward_function_deleter, in_grads, absl::MakeSpan(forward_grads))); in Accumulate()1094 (*forward_function)(in_grads, &forward_grads, use_batch_)); in Accumulate()1096 for (int i = 0; i < forward_grads.size(); ++i) { in Accumulate()1097 if (forward_grads[i] != nullptr) { in Accumulate()1106 vspace_.DeleteGradient(forward_grads[i]); in Accumulate()1108 accumulated_gradients_[output_tensors[i].GetID()] = forward_grads[i]; in Accumulate()