Searched refs:_make_tensor_trace_fun (Results 1 – 1 of 1) sorted by relevance
923 def _make_tensor_trace_fun(self, tensor_name): member in TensorTracer1503 self._make_tensor_trace_fun(tensor_name), processed_out_tensor)1505 trace_fun = self._make_tensor_trace_fun(tensor_name)