Searched refs:trace_tensor (Results 1 – 1 of 1) sorted by relevance
294 def trace_tensor(tensor, tracepoint_name=None): function341 trace_tensor(outputs, '%s' % (checkpoint_name))346 trace_tensor(output_tensor, '%s_%d' % (checkpoint_name, idx))