انضم إلى مجتمع SIG TFX-Addons وساعد في جعل TFX أفضل! انضم إلى SIG TFX-Addons

تينسورفلو :: خدمة:: SavedModelBundleFactory

#include <saved_model_bundle_factory.h>

مصنع يقوم بإنشاء SavedModelBundles من مسارات تصدير SavedModel أو SessionBundle.

ملخص

تدعم الجلسات التي تم إرسالها فقط Run () ، وعلى الرغم من عدم فرضها فمن المتوقع أن يقوم العميل فقط بإجراء مكالمات Run () غير المتغيرة. (إذا كان هذا التقييد ، الذي أضفناه كإجراء أمان ، يمثل مشكلة بالنسبة لحالة الاستخدام الخاصة بك ، فيرجى الاتصال بفريق تقديم TensorFlow لمناقشة تعطيله.)

إذا كانت استدعاءات التكوين للتجميع ، فإن الجلسات المنبعثة تلقائيًا مجموعة مكالمات Run () خلف الكواليس ، باستخدام SharedBatchScheduler المملوك من قبل المصنع. تتم مشاركة سلاسل "config.num_batch_threads" عبر جميع مثيلات الجلسة التي تم إنشاؤها بواسطة هذا المصنع. ومع ذلك ، فإن كل جلسة لها قائمة انتظار خاصة بها من الحجم "config.max_enqueued_batches".

يمكن للمصنع أيضًا تقدير متطلبات الموارد (مثل ذاكرة الوصول العشوائي) لحزمة SavedModelBundle استنادًا إلى SavedModel (أي قبل تحميل الجلسة).

هذه الفئة هي خيط آمن.

وظائف ثابتة عامة

Create (const SessionBundleConfig & config, std::unique_ptr< SavedModelBundleFactory > *factory)
Status
يجسد SavedModelBundleFactory باستخدام ملف config.

الوظائف العامة

CreateSavedModelBundle (const string & path, std::unique_ptr< SavedModelBundle > *bundle)
Status
يتم إنشاء حزمة من مسار تصدير أو SavedModel محدد.
CreateSavedModelBundleWithMetadata (const Loader::Metadata & metadata, const string & path, std::unique_ptr< SavedModelBundle > *bundle)
Status
إنشاء حزمة من مسار تصدير أو SavedModel محدد وبيانات التعريف المحددة.
EstimateResourceRequirement (const string & path, ResourceAllocation *estimate) const
Status
تقدير الموارد التي ستستخدمها حزمة SavedModel بمجرد تحميلها ، من مسار التصدير الخاص بها.
config () const
const SessionBundleConfig &

وظائف ثابتة عامة

يخلق

Status Create(
  const SessionBundleConfig & config,
  std::unique_ptr< SavedModelBundleFactory > *factory
)

يجسد SavedModelBundleFactory باستخدام ملف config.

تفاصيل
المعلمات
config
التكوين مع خيارات التهيئة.
factory
المصنع الذي تم إنشاؤه حديثًا إذا كانت الحالة المرتجعة على ما يرام.

الوظائف العامة

CreateSavedModelBundle

Status CreateSavedModelBundle(
  const string & path,
  std::unique_ptr< SavedModelBundle > *bundle
)

ينشئ حزمة من تصدير محدد أو مسار SavedModel.

تفاصيل
المعلمات
path
الطريق إلى النموذج.
bundle
تم إنشاؤه حديثًا SavedModelBundle إذا كانت الحالة التي تم إرجاعها على ما يرام.

CreateSavedModelBundleWithMetadata

Status CreateSavedModelBundleWithMetadata(
  const Loader::Metadata & metadata,
  const string & path,
  std::unique_ptr< SavedModelBundle > *bundle
)

إنشاء حزمة من مسار تصدير أو SavedModel محدد وبيانات التعريف المحددة.

تفاصيل
المعلمات
metadata
البيانات الوصفية المراد ربطها بالحزمة.
path
الطريق إلى النموذج.
bundle
تم إنشاؤه حديثًا SavedModelBundle إذا كانت الحالة التي تم إرجاعها على ما يرام.

EstimateResourceRequirement

Status EstimateResourceRequirement(
  const string & path,
  ResourceAllocation *estimate
) const 

تقدير الموارد التي ستستخدمها حزمة SavedModel بمجرد تحميلها ، من مسار التصدير الخاص بها.

تفاصيل
المعلمات
path
الطريق إلى النموذج.
estimate
تقديرات استخدام موارد الإخراج. قد يتم تعبئة أنواع مختلفة من الموارد (مثل وحدة المعالجة المركزية وذاكرة الوصول العشوائي وما إلى ذلك).

التكوين

const SessionBundleConfig & config() const