Home
last modified time | relevance | path

Searched refs:gradient_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_gradients_test.py70 w_grad = grad_debugger.gradient_tensor(self.w)
75 w_grad = grad_debugger.gradient_tensor(self.w.name)
80 w_grad = grad_debugger.gradient_tensor(self.w.name)
101 w_grad = grad_debugger.gradient_tensor(self.w)
106 w_grad = grad_debugger.gradient_tensor(self.w.name)
111 w_grad = grad_debugger.gradient_tensor(self.w.name)
138 dz1_dy = grad_debugger_1.gradient_tensor(y)
139 dz2_dy = grad_debugger_2.gradient_tensor(y)
164 grad_debugger_1.gradient_tensor(self.w)
168 grad_debugger_2.gradient_tensor(self.w)
[all …]
H A Ddebug_gradients.py290 gradient_tensor):
299 self._check_same_graph(gradient_tensor)
300 self._gradient_tensors[x_tensor_name] = gradient_tensor
302 def gradient_tensor(self, x_tensor): member in GradientsDebugger
404 gradient_tensor = grad_debugger.gradient_tensor(x_tensor)
406 gradient_tensor.name)
/aosp_15_r20/external/tensorflow/tensorflow/security/advisory/
H A Dtfsa-2021-031.md13 gradient_tensor = tf.constant([0.0], shape=[1])
19 gradients=gradient_tensor, input=input_tensor,