Searched refs:tensor_trace_points (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer_report.py | 246 tensor_trace_points, collected_signature_types): argument
291 if tensor.name in tensor_trace_points:
292 tensor_def.trace_point_name = tensor_trace_points[tensor.name]
313 tensor_trace_order, tensor_trace_points): argument
319 self._write_trace_points(tensor_trace_points)
324 def _write_trace_points(self, tensor_trace_points): argument
328 for (tensor, checkpoint_name) in tensor_trace_points:
|
| D | tensor_tracer.py | 1180 tensor_trace_points, argument
1199 for (tensor, _) in tensor_trace_points])
1255 tensor_trace_points = graph.get_collection(_TENSOR_TRACER_COLLECTION)
1259 graph_order, ops_in_exec_path, tensor_trace_points, report_handler)
1279 tensor_trace_points, self._signature_types())
1291 tensor_trace_order, tensor_trace_points)
|