Searched refs:grad_debugger (Results 1 – 2 of 2) sorted by relevance
59 grad_debugger = debug_gradients.GradientsDebugger()60 id_grad_w = grad_debugger.identify_gradient(self.w)74 w_grad = grad_debugger.gradient_tensor(self.w)79 w_grad = grad_debugger.gradient_tensor(self.w.name)84 w_grad = grad_debugger.gradient_tensor(self.w.name)89 grad_debugger = debug_gradients.GradientsDebugger()90 id_grad_w = grad_debugger.identify_gradient(self.w)93 with grad_debugger:105 w_grad = grad_debugger.gradient_tensor(self.w)110 w_grad = grad_debugger.gradient_tensor(self.w.name)[all …]
361 grad_debugger = _gradient_debuggers[grad_debugger_uuid]362 grad_debugger.register_gradient_tensor(orig_tensor_name, dy)372 def gradient_values_from_dump(grad_debugger, x_tensor, dump): argument403 if (dump.python_graph and grad_debugger.graph and404 dump.python_graph != grad_debugger.graph):408 (grad_debugger.graph, dump.python_graph))410 gradient_tensor = grad_debugger.gradient_tensor(x_tensor)