Searched refs:gradient_tensor (Results 1 – 2 of 2) sorted by relevance
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)105 w_grad = grad_debugger.gradient_tensor(self.w)110 w_grad = grad_debugger.gradient_tensor(self.w.name)115 w_grad = grad_debugger.gradient_tensor(self.w.name)142 dz1_dy = grad_debugger_1.gradient_tensor(y)143 dz2_dy = grad_debugger_2.gradient_tensor(y)168 grad_debugger_1.gradient_tensor(self.w)172 grad_debugger_2.gradient_tensor(self.w)[all …]
296 gradient_tensor):305 self._check_same_graph(gradient_tensor)306 self._gradient_tensors[x_tensor_name] = gradient_tensor308 def gradient_tensor(self, x_tensor): member in GradientsDebugger410 gradient_tensor = grad_debugger.gradient_tensor(x_tensor)412 gradient_tensor.name)