Searched refs:gradient_tensors (Results 1 – 4 of 4) sorted by relevance
331 def gradient_tensors(self): member in GradientsDebugger
277 self.assertEqual(2, len(grad_debugger.gradient_tensors()))299 self.assertEqual(3, len(grad_debugger.gradient_tensors()))
1175 tensorflow::gtl::ArraySlice<PyObject*> gradient_tensors) const final { in AggregateGradients()1176 PyObject* list = PyList_New(gradient_tensors.size()); in AggregateGradients()1177 for (int i = 0; i < gradient_tensors.size(); ++i) { in AggregateGradients()1179 CHECK(gradient_tensors[i] != nullptr); in AggregateGradients()1180 CHECK(gradient_tensors[i] != Py_None); in AggregateGradients()1182 reinterpret_cast<PyObject*>(gradient_tensors[i])); in AggregateGradients()
92 gtl::ArraySlice<Gradient*> gradient_tensors) const = 0;