TPUReplicateMetadata

الطبقة النهائية العامة TPUReplicateMetadata

البيانات الوصفية التي تشير إلى كيفية تكرار حساب TPU.

تحتوي هذه العملية على البيانات الوصفية المشتركة في عمليات الرسم البياني الفرعي للحساب `tpu.replicate()`.

فئات متداخلة

فصل TPUReplicateMetadata.Options السمات الاختيارية لـ TPUReplicateMetadata

الأساليب العامة

ثابت TPUReplicateMetadata.Options
allowSoftPlacement (المنطقيةallowSoftPlacement)
ثابت TPUReplicateMetadata.Options
شكل الحساب (شكل الحساب في القائمة <الطويلة)
ثابت TPUReplicateMetadata
إنشاء ( نطاق النطاق ، numReplicas طويلة، خيارات... خيارات)
طريقة المصنع لإنشاء فئة تغلف عملية TPUReplicateMetadata جديدة.
ثابت TPUReplicateMetadata.Options
تعيين الجهاز (قائمة<طويلة> تعيين الجهاز)
ثابت TPUReplicateMetadata.Options
hostComputeCore (قائمة<String> hostComputeCore)
ثابت TPUReplicateMetadata.Options
numCoresPerReplica (Long numCoresPerReplica)
ثابت TPUReplicateMetadata.Options
خريطة الحشو (قائمة <سلسلة> خريطة الحشو)
ثابت TPUReplicateMetadata.Options
stepMarkerLocation (سلسلة stepMarkerLocation)
ثابت TPUReplicateMetadata.Options
طوبولوجيا (طوبولوجيا السلسلة)
ثابت TPUReplicateMetadata.Options
tpuCompileOptionsProto (سلسلة tpuCompileOptionsProto)
ثابت TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (الاستخدام المنطقي SpmdForXlaPartitioning)
ثابت TPUReplicateMetadata.Options
useTpu (استخدام منطقي)

الطرق الموروثة

الأساليب العامة

الثابتة العامة TPUReplicateMetadata.OptionsallowSoftPlacement ( المنطقيةallowSoftPlacement)

ثابت عام TPUReplicateMetadata.Options computationShape (List<Long> computationShape)

حدود
computationShape إهمال. استخدم num_cores_per_replica بدلاً من ذلك.

إنشاء بيانات TPUReplicateMetadata الثابتة العامة ( نطاق النطاق ، الأرقام الطويلة، الخيارات... الخيارات)

طريقة المصنع لإنشاء فئة تغلف عملية TPUReplicateMetadata جديدة.

حدود
نِطَاق النطاق الحالي
numReplicas عدد النسخ المتماثلة للحساب
خيارات يحمل قيم السمات الاختيارية
عائدات
  • مثيل جديد لـ TPUReplicateMetadata

الثابت العام لـ TPUReplicateMetadata.Options devicesAssignment (قائمة<Long>deviceAssignment)

حدود
DeviceAssignment تعيين الأجهزة للحساب.

TPUReplicateMetadata.Options العام الثابت hostComputeCore (List<String> hostComputeCore)

ثابت عام TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

حدود
numCoresPerReplica عدد النوى لكل نسخة متماثلة. تستخدم للتوازي النموذجي.

ثابت عام TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

ثابت عام TPUReplicateMetadata.Options stepMarkerLocation (سلسلة stepMarkerLocation)

طوبولوجيا TPUReplicateMetadata.Options العامة الثابتة (طوبولوجيا السلسلة)

حدود
البنية يشير TopologyProto إلى طوبولوجيا شريحة جراب TPU.

ثابت عام TPUReplicateMetadata.Options tpuCompileOptionsProto (سلسلة tpuCompileOptionsProto)

الثابت العام لـ TPUReplicateMetadata.Options useSpmdForXlaPartitioning (الاستخدام المنطقي SpmdForXlaPartitioning)

TPUReplicateMetadata.Options العام الثابت useTpu (المنطقي useTpu)

حدود
com.useTpu ما إذا كان سيتم وضع الحساب على TPU.