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