In questo momento il miglior modo esterno è compilare con un profiler CPU/GPU, ma questo deve essere fatto manualmente modificando le opzioni nei file BUILD
e tensorflow.bzl
(non sicuro dove).
È possibile quindi ottenere informazioni sul profilo su quali funzioni stanno prendendo più tempo, ecc E 'a voi, è possibile eseguire utilizzando gperftools e la visualizzazione da pprof
Dal codice di tensorflow è collegata tramite swig
, i' Non sono sicuro al 100% quanto sia facile ottenere i simboli durante la lettura del profilo tramite pprof
. Potrebbe essere possibile indicarlo al file di swag .so
che è stato creato.
Provalo e facci sapere se funziona per te!
fonte
2016-03-09 20:48:35
Possibile duplicato: http://stackoverflow.com/questions/34293714/tensorflow-can-i-measure-the-execution-time-of-individual-operations – user3559888