本文整理匯總了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)