Searched refs:tpu_wrap_trace_fn (Results 1 – 1 of 1) sorted by relevance
1847 def tpu_wrap_trace_fn(tensor, out_tensor_name): function1863 trace_op = tpu_wrap_trace_fn(processed_out_tensor, tensor_name)