本文整理汇总了Python中tensorflow.python.ops.check_ops方法的典型用法代码示例。如果您正苦于以下问题：Python ops.check_ops方法的具体用法？Python ops.check_ops怎么用？Python ops.check_ops使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.python.ops的用法示例。
在下文中一共展示了ops.check_ops方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: _call_model_fn
# 需要导入模块: from tensorflow.python import ops [as 别名]
# 或者: from tensorflow.python.ops import check_ops [as 别名]
def _call_model_fn(self, features, labels, mode, config):
if mode == _INFERENCE_ON_TPU_MODE:
context = tpu._TPUInferenceContext('tpu_inference', check_ops=False)
try:
context.Enter()
if (
(self._export_saved_model_api_version ==
ExportSavedModelApiVersion.V1)
or self._export_saved_model_api_version == 1):
result = self._call_model_fn_for_inference(features, labels, mode,
config)
else:
result = super(TPUEstimator,
self)._call_model_fn(features, labels, mode, config)
finally:
context.Exit()
return result
else:
return super(TPUEstimator, self)._call_model_fn(features, labels, mode,
config)