當前位置: 首頁>>代碼示例>>Python>>正文


Python ops.check_ops方法代碼示例

本文整理匯總了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) 
開發者ID:tensorflow,項目名稱:estimator,代碼行數:22,代碼來源:tpu_estimator.py


注:本文中的tensorflow.python.ops.check_ops方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。