Searched refs:trace_cpu (Results 1 – 2 of 2) sorted by relevance
1598 def trace_cpu(self, graph, tensor_fetches, op_fetches=None): member in TensorTracer
363 loss = tt.trace_cpu(ops.get_default_graph(), loss, self.train_op)