Home
last modified time | relevance | path

Searched refs:out_gradients (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dtape.h525 std::vector<Gradient*> out_gradients; in ComputeGradient() local
526 out_gradients.reserve(trace.output_tensor_info.size()); in ComputeGradient()
536 out_gradients.push_back(nullptr); in ComputeGradient()
538 out_gradients.push_back(vspace.Zeros(trace.output_tensor_info[i])); in ComputeGradient()
555 out_gradients.push_back(new_gradients); in ComputeGradient()
561 out_gradients, &in_gradients); in ComputeGradient()
574 for (Gradient* grad : out_gradients) { in ComputeGradient()