Searched defs:tensor_trace_order (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tensor_tracer_report.py | 235 def create_report_proto(self, tt_config, tt_parameters, tensor_trace_order, argument
315 tensor_trace_order, tensor_trace_points): argument
413 def _write_cache_index_map_section(self, tensor_trace_order): argument
|
| H A D | tensor_tracer.py | 962 def _make_tensor_trace_fun(self, tensor_name, tensor_trace_order): argument
1392 tensor_trace_order): argument
1467 tensor_trace_order, graph): argument
1569 tensor_trace_order, graph): argument
|