Searched refs:SumGradients (Results 1 – 2 of 2) sorted by relevance
201 NodeOut SumGradients(const NodeOut& src);294 NodeOut SymbolicGradientBuilder::SumGradients(const NodeOut& src) { in SumGradients() function in tensorflow::SymbolicGradientBuilder360 dy[i] = SumGradients({n, i}); in Compute()392 (*x_grad_node_outputs_)[i] = SumGradients(x_node_outputs_[i]); in Compute()
72 Status SumGradients(const Output& src, Output* grad);342 Status SymbolicGradientBuilder::SumGradients(const Output& src, Output* grad) { in SumGradients() function in tensorflow::__anon2882cf250111::SymbolicGradientBuilder454 TF_RETURN_IF_ERROR(SumGradients({n, i}, &dy[i])); in AddGradients()544 TF_RETURN_IF_ERROR(SumGradients(nout, &(*grad_outputs_)[idx])); in AddGradients()