hub / github.com/alibaba/nann / trace_tpu

Method trace_tpu

tensorflow/tensorflow/python/tpu/tensor_tracer.py:1444–1501 · view source on GitHub ↗

Traces the tensors generated by TPU Ops in a TF graph. Args: graph: the graph of Ops executed on the TPU. tensor_fetches: a (list,tuple,or a single object) of tensor fetches returned by model_fn given to session.run. Function must be provided with as least one tensor

(self, graph,
                tensor_fetches,
                op_fetches=None,
                num_replicas=None,
                num_replicas_per_host=None,
                num_hosts=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

split_compile_and_replicateFunction · 0.95

body_wrapperFunction · 0.95

Calls 4

_add_replica_id_to_graphMethod · 0.95

_trace_executionMethod · 0.95

addMethod · 0.45

as_defaultMethod · 0.45

Tested by

no test coverage detected