配置文件摘要(按操作类型)

时间:2019-07-10 14:05:58

标签: tensorflow2.0

我正在Tensorflow 2.0中分析我的模型。我可以按照功能说明使用Tensorboard获取配置文件和视图。我使用tf.summary.trace_on和tf.summary.trace_export: https://www.tensorflow.org/tensorboard/r2/graphs

现在,我想按操作类型创建一个摘要,类似于tflite探查器(https://github.com/tensorflow/tensorflow/tree/master/tensorflow/lite/tools/benchmark)中的节点类型摘要。

TensorFlow 1.14 tf.profile.Profiler看起来能够使用'cmd'选项进行操作: https://www.tensorflow.org/api_docs/python/tf/profiler/profile

如何使用TensorFlow 2.0做到这一点?

0 个答案:

没有答案