tf.config.optimizer.set_jit
| Просмотреть исходный код на GitHub |
Устанавливает, включена ли JIT-компиляция.
tf.config.optimizer.set_jit(
enabled
)
Обратите внимание, что оптимизации применяются только к коду, который компилируется в граф. В режиме eager, который является значением по умолчанию для API TF2, это означает только код, определённый под декоратором tf.function.
| Аргументы | |
|---|---|
enabled | Включить JIT-компиляцию. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/config/optimizer/set_jit