Searched refs:tensor_trace_points (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tensor_tracer_report.py | 236 tensor_trace_points, collected_signature_types): argument
281 if tensor.name in tensor_trace_points:
282 tensor_def.trace_point_name = tensor_trace_points[tensor.name]
315 tensor_trace_order, tensor_trace_points): argument
321 self._write_trace_points(tensor_trace_points)
326 def _write_trace_points(self, tensor_trace_points): argument
330 for (tensor, checkpoint_name) in tensor_trace_points:
|
| H A D | tensor_tracer.py | 1265 tensor_trace_points, argument
1284 for (tensor, _) in tensor_trace_points])
1345 tensor_trace_points = graph.get_collection(_TENSOR_TRACER_COLLECTION)
1349 graph_order, ops_in_exec_path, tensor_trace_points, report_handler)
1369 tensor_trace_points, self._signature_types())
1383 tensor_trace_order, tensor_trace_points)
|