Searched refs:in_gradients (Results 1 – 1 of 1) sorted by relevance
558 std::vector<Gradient*> in_gradients; in ComputeGradient() local561 out_gradients, &in_gradients); in ComputeGradient()570 in_gradients.resize(trace.input_tensor_id.size()); in ComputeGradient()580 VLOG(1) << "Got " << in_gradients.size() << " in_gradients for " in ComputeGradient()582 for (int i = 0; i < in_gradients.size(); ++i) { in ComputeGradient()584 if (in_gradients[i] != nullptr) { in ComputeGradient()586 unaggregated_grads.push_back(in_gradients[i]); in ComputeGradient()