Searched refs:traced_tensors (Results 1 – 2 of 2) sorted by relevance
150 def __init__(self, graph_order, traced_tensors): argument152 self.traced_tensors = traced_tensors159 for out_tensor in self.traced_tensors:
1197 traced_tensors = []1208 traced_tensors.append(out_tensor)1209 return traced_tensors1258 traced_tensors = self._determine_and_instrument_traced_tensors(1260 logging.info('TensorTracer is tracing %d tensors.', len(traced_tensors))1263 traced_tensors)1268 self._create_temp_cache(len(traced_tensors), num_signatures)1272 [len(traced_tensors),