ReplicateMetadata

کلاس نهایی عمومی 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 allowSoftPlacement (Bolean allowSoftPlacement)

عمومی استاتیک ReplicateMetadata.Options computationShape (List<Long> computationShape)

مولفه های
محاسبه شکل منسوخ. به جای آن از num_cores_per_replica استفاده کنید.

ایجاد ReplicateMetadata ایستا عمومی ( حوزه دامنه ، تعداد Replicas طولانی، گزینه‌ها... گزینه‌ها)

روش کارخانه برای ایجاد کلاسی که یک عملیات ReplicateMetadata جدید را بسته بندی می کند.

مولفه های
محدوده محدوده فعلی
numReplicas تعداد کپی های محاسبات
گزینه ها مقادیر ویژگی های اختیاری را حمل می کند
برمی گرداند
  • یک نمونه جدید از ReplicateMetadata

عمومی استاتیک ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

مولفه های
تخصیص دستگاه تخصیص دستگاه ها برای محاسبات

عمومی static ReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

عمومی استاتیک ReplicateMetadata.Options numCoresPerReplica (Long NumCoresPerReplica)

مولفه های
numCoresPerReplica تعداد هسته در هر ماکت برای موازی سازی مدل استفاده می شود.

عمومی استاتیک ReplicateMetadata.Options paddingMap (List<String> paddingMap)

عمومی استاتیک ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

توپولوژی استاتیک عمومی ReplicateMetadata.Options (توپولوژی رشته ای)

مولفه های
توپولوژی TopologyProto توپولوژی برش غلاف TPU را نشان می دهد.

عمومی استاتیک ReplicateMetadata.Options useSpmdForXlaPartitioning (استفاده بولیSpmdForXlaPartitioning)

عمومی استاتیک ReplicateMetadata.Options useTpu (Boolean useTpu)

مولفه های
useTpu اینکه آیا محاسبات را روی TPU قرار دهیم.