Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_gradients.py212 tensor_name_regex = []
214 tensor_name_regex.append(re.escape(tensor.name) + "$")
215 tensor_name_regex = "(" + "|".join(tensor_name_regex) + ")"
216 return self.watch_gradients_by_tensor_names(graph, tensor_name_regex)
218 def watch_gradients_by_tensor_names(self, graph, tensor_name_regex): argument
261 tensor_name_pattern = re.compile(tensor_name_regex)