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()))
1031 tensorflow::gtl::ArraySlice<PyObject*> gradient_tensors) const final { in AggregateGradients()1032 PyObject* list = PyList_New(gradient_tensors.size()); in AggregateGradients()1033 for (int i = 0; i < gradient_tensors.size(); ++i) { in AggregateGradients()1035 CHECK(gradient_tensors[i] != nullptr); in AggregateGradients()1036 CHECK(gradient_tensors[i] != Py_None); in AggregateGradients()1038 reinterpret_cast<PyObject*>(gradient_tensors[i])); in AggregateGradients()
86 gtl::ArraySlice<Gradient*> gradient_tensors) const = 0;