Searched refs:tensor_name_regex (Results 1 – 1 of 1) sorted by relevance
218 tensor_name_regex = []220 tensor_name_regex.append(re.escape(tensor.name) + "$")221 tensor_name_regex = "(" + "|".join(tensor_name_regex) + ")"222 return self.watch_gradients_by_tensor_names(graph, tensor_name_regex)224 def watch_gradients_by_tensor_names(self, graph, tensor_name_regex): argument267 tensor_name_pattern = re.compile(tensor_name_regex)