کلاس نهایی عمومی TPUReplicateMetadata
فراداده نشان می دهد که چگونه محاسبات TPU باید تکرار شود.
این عملیات ابرداده مشترک عملیات یک زیرگراف محاسبه «tpu.replicate()» را نگه می دارد.
کلاس های تو در تو
| کلاس | TPUReplicateMetadata.Options | ویژگی های اختیاری برای TPUReplicateMetadata | |
روش های عمومی
| استاتیک TPUReplicateMetadata.Options | allowSoftPlacement (Bolean allowSoftPlacement) |
| استاتیک TPUReplicateMetadata.Options | computationShape (List<Long> computationShape) |
| استاتیک TPUReplicate Metadata | ایجاد (حوزه دامنه ، تعداد طولانی Replicas، گزینه‌ها... گزینه‌ها) روش کارخانه برای ایجاد کلاسی که عملیات TPUReplicateMetadata جدید را بسته بندی می کند. |
| استاتیک TPUReplicateMetadata.Options | deviceAssignment (List<Long> deviceAssignment) |
| استاتیک TPUReplicateMetadata.Options | hostComputeCore (List<String> hostComputeCore) |
| استاتیک TPUReplicateMetadata.Options | numCoresPerReplica (NumCoresPerReplica طولانی) |
| استاتیک TPUReplicateMetadata.Options | paddingMap (List<String> paddingMap) |
| استاتیک TPUReplicateMetadata.Options | stepMarkerLocation (رشته stepMarkerLocation) |
| استاتیک TPUReplicateMetadata.Options | توپولوژی (توپولوژی رشته ای) |
| استاتیک TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (استفاده بولیSpmdForXlaPartitioning) |
| استاتیک TPUReplicateMetadata.Options | useTpu (Boolean useTpu) |
روش های ارثی
روش های عمومی
عمومی ثابت TPUReplicateMetadata. Options computationShape (List<Long> computationShape)
مولفه های
| محاسبه شکل | منسوخ. به جای آن از num_cores_per_replica استفاده کنید. |
|---|
ایجاد عمومی ایستا TPUReplicateMetadata ( حوزه دامنه ، تعداد Replicas طولانی، گزینه‌ها... گزینه‌ها)
روش کارخانه برای ایجاد کلاسی که عملیات TPUReplicateMetadata جدید را بسته بندی می کند.
مولفه های
| محدوده | محدوده فعلی |
|---|---|
| numReplicas | تعداد کپی های محاسبات |
| گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از TPUReplicateMetadata
عمومی استاتیک TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
مولفه های
| تخصیص دستگاه | تخصیص دستگاه ها برای محاسبات |
|---|
عمومی ثابت TPUReplicateMetadata.Options numCoresPerReplica (NumCoresPerReplica طولانی)
مولفه های
| numCoresPerReplica | تعداد هسته در هر ماکت برای موازی سازی مدل استفاده می شود. |
|---|
توپولوژی ایستا عمومی TPUReplicateMetadata.Options (توپولوژی رشته ای)
مولفه های
| توپولوژی | TopologyProto توپولوژی برش غلاف TPU را نشان می دهد. |
|---|
عمومی ثابت TPUReplicateMetadata.Options useSpmdForXlaPartitioning (استفاده بولیSpmdForXlaPartitioning)
عمومی ثابت TPUReplicateMetadata.Options useTpu (Boolean useTpu)
مولفه های
| useTpu | اینکه آیا محاسبات را روی TPU قرار دهیم. |
|---|