Searched refs:tensor_trace_order (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer_report.py | 245 def create_report_proto(self, tt_config, tt_parameters, tensor_trace_order, argument
276 for tensor in tensor_trace_order.graph_order.tensors:
279 if tensor.name in tensor_trace_order.tensorname_to_cache_idx:
282 tensor_trace_order.tensorname_to_cache_idx[tensor.name])
301 tf_graph = tensor_trace_order.graph_order.graph
313 tensor_trace_order, tensor_trace_points): argument
317 self._write_op_list_section(tensor_trace_order.graph_order)
318 self._write_tensor_list_section(tensor_trace_order.graph_order)
320 self._write_cache_index_map_section(tensor_trace_order)
322 self._write_graph_section(tensor_trace_order.graph_order)
[all …]
|
| D | tensor_tracer.py | 877 def _make_tensor_trace_fun(self, tensor_name, tensor_trace_order): argument
908 if tensor_name not in tensor_trace_order.tensorname_to_cache_idx:
912 msg = '%d' % tensor_trace_order.tensorname_to_cache_idx[tensor_name]
1262 tensor_trace_order = tensor_tracer_report.TensorTraceOrder(graph_order,
1278 self._tt_config, self._parameters, tensor_trace_order,
1291 tensor_trace_order, tensor_trace_points)
1292 return tensor_trace_order
1300 tensor_trace_order): argument
1350 tensor_trace_order.tensorname_to_cache_idx.items(),
1369 def _generate_flush_cache_op(self, num_replicas, on_tpu, tensor_trace_order): argument
[all …]
|