projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ce8e19a) | patch
HLO profiling for tfcompile.
authorSanjoy Das <sanjoy@google.com>
Sat, 28 Apr 2018 03:06:35 +0000 (20:06 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sat, 28 Apr 2018 03:09:38 +0000 (20:09 -0700)
commit74747435c2442084e8de53bc73311152f270ae88
treed7d9a57b195039d277fc8af9814290a787d04cc4tree | snapshot
parentce8e19a756f71fa66f60a28515c64c106ca7f6a1commit | diff
HLO profiling for tfcompile.

This CL extends the --xla_hlo_profile knob to tfcompile.  tf_library rules can
now set enable_xla_hlo_profiling to True to:

 - Have the generated code update per-HLO profile counters as it executes.
 - Have tfcompile generate and serialize an instance HloProfilePrinterData with
   a compiled model that can be used to pretty-print the collected profile
   counters.

PiperOrigin-RevId: 194627272
13 files changed:
tensorflow/compiler/aot/codegen.cc diff | blob | history
tensorflow/compiler/aot/codegen.h diff | blob | history
tensorflow/compiler/aot/codegen_test.cc diff | blob | history
tensorflow/compiler/aot/codegen_test_h.golden diff | blob | history
tensorflow/compiler/aot/compile.cc diff | blob | history
tensorflow/compiler/aot/embedded_protocol_buffers.cc diff | blob | history
tensorflow/compiler/aot/embedded_protocol_buffers.h diff | blob | history
tensorflow/compiler/aot/tests/BUILD diff | blob | history
tensorflow/compiler/aot/tests/tfcompile_test.cc diff | blob | history
tensorflow/compiler/aot/tfcompile.bzl diff | blob | history
tensorflow/compiler/aot/tfcompile_main.cc diff | blob | history
tensorflow/compiler/xla/service/cpu/cpu_compiler.cc diff | blob | history
tensorflow/compiler/xla/service/cpu/cpu_compiler.h diff | blob | history
Domain: Machine Learning / ML Framework;