В TensorFlow Eager я могу использовать профилировщик Python для профилирования кода, который работает исключительно в нетерпеливом режиме.Однако, если я «скомпилирую» функцию python в графическую функцию, используя tf.function
или tf.contrib.eager.defun
, эта функция станет непрозрачной для python - профилировщик не сможет ее ввести.
Я узнал, как профилироватьГрафик TF в графическом режиме, но я не знаю, как это сделать с помощью графической функции в активном режиме.
В частности, если я создаю такую функцию,
tf.enable_v2_behavior()
@tf.function
def myfunc(x):
y = op1(x)
z = op2(y, z)
return z
какузнать, сколько времени тратится на op1
и op2
при выполнении myfunc?