Searched refs:grad_outputs_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients_test.cc | 50 AddSymbolicGradients(scope_, outputs_, inputs_, &grad_outputs_)); in CreateBackprop()
51 ASSERT_EQ(grad_outputs_.size(), inputs_.size()); in CreateBackprop()
74 TF_ASSERT_OK(session.Run(run_options, feeds, grad_outputs_, run_outputs, in Run()
76 ASSERT_EQ(out_tensors.size(), grad_outputs_.size()); in Run()
88 std::vector<Output> grad_outputs_; member in tensorflow::__anon97a6b1da0111::WhileGradientsTest
213 AddSymbolicGradients(cpu1_scope, outputs_, inputs_, &grad_outputs_)); in TEST_F()
214 ASSERT_EQ(grad_outputs_.size(), inputs_.size()); in TEST_F()
|
| D | gradients.cc | 107 std::vector<Output>* grad_outputs_; member in tensorflow::__anonc8091ec30111::SymbolicGradientBuilder
150 grad_outputs_(grad_outputs) {} in SymbolicGradientBuilder()
264 grad_outputs_->clear(); in Initialize()
265 grad_outputs_->resize(inputs_.size()); in Initialize()
462 (*grad_outputs_)[iter->second] = dy[i]; in AddGradients()
544 DCHECK(((*grad_outputs_)[idx].node() == nullptr)); in AddGradients()
545 TF_RETURN_IF_ERROR(SumGradients(nout, &(*grad_outputs_)[idx])); in AddGradients()
|