TPUReplicateMetadata

کلاس نهایی عمومی TPUReplicateMetadata

فراداده نشان می دهد که چگونه محاسبات TPU باید تکرار شود.

این عملیات ابرداده مشترک عملیات یک زیرگراف محاسبه «tpu.replicate()» را نگه می دارد.

کلاس های تو در تو

کلاس TPUReplicateMetadata.Options ویژگی های اختیاری برای TPUReplicateMetadata

ثابت ها

رشته OP_NAME نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

روش های عمومی

استاتیک 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)

روش های ارثی

ثابت ها

رشته نهایی ثابت عمومی OP_NAME

نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

مقدار ثابت: "TPUReplicateMetadata"

روش های عمومی

عمومی ایستا TPUReplicateMetadata.Options allowSoftPlacement (Bolean allowSoftPlacement)

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

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

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

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

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

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

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

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

عمومی ثابت TPUReplicateMetadata.Options numCoresPerReplica (NumCoresPerReplica طولانی)

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

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

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

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

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

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

عمومی ثابت TPUReplicateMetadata.Options useTpu (Boolean useTpu)

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