Searched refs:out_tensor_name (Results 1 – 1 of 1) sorted by relevance
1847 def tpu_wrap_trace_fn(tensor, out_tensor_name): argument1849 tensor_trace_fn = self._make_tensor_trace_fun(out_tensor_name,