Searched refs:trace_tpu (Results 1 – 3 of 3) sorted by relevance
168 output_tensors = tt.trace_tpu(ops.get_default_graph(),
919 output_tensors = tt.trace_tpu(ops.get_default_graph(),
1537 def trace_tpu(self, graph, member in TensorTracer