jax.export.DisabledSafetyCheck

class jax.export.DisabledSafetyCheck(_impl)

在 (反)序列化时应跳过的安全检查。

大多数这些检查在序列化时执行，但有些被推迟到反序列化。禁用检查的列表附加到序列化中，例如，作为 jax.export.Exported 的字符串属性序列或 tf.XlaCallModuleOp。

使用 jax2tf 时，可以通过传递 TF_XLA_FLAGS=–tf_xla_call_module_disabled_checks=platform 来禁用更多反序列化安全检查。

参数：

_impl (str)

__init__(_impl)

参数：

_impl (str)

方法

__init__(_impl)
custom_call(target_name)	允许序列化一个未知是否稳定的调用目标。
is_custom_call()	返回此指令允许的自定义调用目标。
platform()	允许编译平台与导出平台不同。
shape_assertions()	已弃用：无操作。