org.tensorflow.op.tpu

ชั้นเรียน

AllToAll <T ขยาย TType > Op เพื่อแลกเปลี่ยนข้อมูลระหว่างแบบจำลอง TPU
CollectivePermute <T ขยาย TType > Op เพื่อเปลี่ยนเทนเซอร์ข้ามอินสแตนซ์ TPU ที่จำลองแบบ
ผลการรวบรวม ส่งกลับผลลัพธ์ของการคอมไพล์ TPU
คอมไพล์สำเร็จยืนยัน ยืนยันว่าการรวบรวมสำเร็จ
กำหนดค่า DistributedTPU ตั้งค่าโครงสร้างแบบรวมศูนย์สำหรับระบบ TPU แบบกระจาย
กำหนดค่า DistributedTPU.Options แอ็ตทริบิวต์ทางเลือกสำหรับ ConfigureDistributedTPU
กำหนดค่าการฝัง TPU ตั้งค่า TPUEmbedding ในระบบ TPU แบบกระจาย
CrossReplicaSum <T ขยาย TNumber > อินพุต Op to sum ในอินสแตนซ์ TPU ที่จำลองแบบ
การฝังการเปิดใช้งาน การดำเนินการที่ทำให้เกิดความแตกต่างของ TPU Embeddings
เข้าคิว TPUembedingIntegerBatch การดำเนินการที่จัดคิวรายการเทนเซอร์แบทช์อินพุตเป็น TPUEmbedding
เข้าคิว TPUmbeddingIntegerBatch.Options แอตทริบิวต์เพิ่มเติมสำหรับ EnqueueTPUEmbeddingIntegerBatch
เข้าคิว TPUembedRaggedTensorBatch ทำให้การย้ายโค้ดที่ใช้ tf.nn.embedding_lookup() ง่ายขึ้น
เข้าคิว TPUmbeddingRaggedTensorBatch.Options แอตทริบิวต์ทางเลือกสำหรับ EnqueueTPUEmbeddingRaggedTensorBatch
เข้าคิวTPUmbeddingSparseBatch การดำเนินการที่จัดคิวดัชนีอินพุต TPUEmbedding จาก SparseTensor
จัดคิวTPUmbeddingSparseBatch.Options แอตทริบิวต์เพิ่มเติมสำหรับ EnqueueTPUEmbeddingSparseBatch
จัดคิว TPUembedSparseTensorBatch ทำให้การย้ายโค้ดที่ใช้ tf.nn.embedding_lookup_sparse() ง่ายขึ้น
จัดคิวTPUmbeddingSparseTensorBatch.Options แอตทริบิวต์เพิ่มเติมสำหรับ EnqueueTPUEmbeddingSparseTensorBatch
ดำเนินการ Op ที่โหลดและรันโปรแกรม TPU บนอุปกรณ์ TPU
ExecuteAndUpdateVariables Op ที่รันโปรแกรมด้วยการอัพเดตตัวแปรแบบแทนที่ซึ่งเป็นตัวเลือก
InfeedDequeue <T ขยาย TType > ตัวยึดตำแหน่งใช้สำหรับค่าที่จะป้อนเข้าสู่การคำนวณ
InfeedDequeueTuple ดึงค่าหลายค่าจากการป้อนเข้าเป็นทูเพิล XLA
ป้อนเข้าคิว op ที่ป้อนค่า Tensor เดียวในการคำนวณ
InfeedEnqueue.ตัวเลือก แอตทริบิวต์เพิ่มเติมสำหรับ InfeedEnqueue
InfeedEnqueuePrelinearizedBuffer การดำเนินการที่จัดคิวบัฟเฟอร์ที่กำหนดไว้ล่วงหน้าลงในอินพุต TPU
InfeedEnqueuePrelinearizedBuffer.Options แอ็ตทริบิวต์ทางเลือกสำหรับ InfeedEnqueuePrelinearizedBuffer
InfeedEnqueueTuple ฟีดค่าเทนเซอร์หลายค่าลงในการคำนวณเป็นทูเพิล XLA
InfeedEnqueueTuple.Options แอ็ตทริบิวต์ทางเลือกสำหรับ InfeedEnqueueTuple
โหลดTPUEmbeddingAdadeltaParameters โหลดพารามิเตอร์การฝัง Adadelta
โหลดTPUEmbeddingAdadeltaParameters.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingAdadeltaParameters
โหลด TPU การฝัง AdadeltaParametersGradAccumDebug โหลดพารามิเตอร์ Adadelta พร้อมการสนับสนุนการดีบัก
โหลด TPU การฝัง AdadeltaParametersGradAccumDebug.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingAdadeltaParametersGradAccumDebug
โหลดTPUEmbeddingAdagradParameters โหลดพารามิเตอร์การฝัง Adagrad
โหลดTPUEmbeddingAdagradParameters.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingAdagradParameters
โหลด TPU การฝัง AdagradParametersGradAccumDebug โหลดพารามิเตอร์การฝัง Adagrad ด้วยการสนับสนุนการแก้ไขข้อบกพร่อง
โหลดTPUEmbeddingAdagradParametersGradAccumDebug.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingAdagradParametersGradAccumDebug
โหลดTPUEmbeddingADAMพารามิเตอร์ โหลดพารามิเตอร์การฝัง ADAM
โหลดTPUEmbeddingADAMParameters.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingADAMParameters
โหลด TPU การฝัง ADAMP พารามิเตอร์ GradAccum Debug โหลดพารามิเตอร์การฝัง ADAM พร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
โหลดTPUEmbeddingADAMParametersGradAccumDebug.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingADAMParametersGradAccumDebug
โหลดTPUEmbeddingCenteredRMSPropParameters โหลดพารามิเตอร์การฝัง RMSProp ที่กึ่งกลาง
LoadTPUEmbeddingCenteredRMSPropParameters.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingCenteredRMSPropParameters
โหลดTPUEmbeddingFTRLParameters โหลดพารามิเตอร์การฝัง FTRL
โหลดTPUEmbeddingFTRLParameters.Options คุณลักษณะเพิ่มเติมสำหรับ LoadTPUEmbeddingFTRLParameters
โหลด TPU การฝัง FTRLParametersGradAccumDebug โหลดพารามิเตอร์การฝัง FTRL พร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
โหลดTPUEmbeddingFTRLParametersGradAccumDebug.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingFTRLParametersGradAccumDebug
โหลดTPUEmbeddingMDLAdagradLightParameters โหลดพารามิเตอร์การฝัง MDL Adagrad Light
โหลดTPUEmbeddingMDLAdagradLightParameters.Options คุณลักษณะเพิ่มเติมสำหรับ LoadTPUEmbeddingMDLAdagradLightParameters
โหลดพารามิเตอร์ TPU การฝังโมเมนตัม โหลดพารามิเตอร์การฝังโมเมนตัม
โหลดTPUEmbeddingMomentumParameters.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingMomentumParameters
โหลด TPU การฝังโมเมนตัมพารามิเตอร์ GradAccum Debug โหลดพารามิเตอร์การฝังโมเมนตัมพร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
โหลด TPU การฝังโมเมนตัมพารามิเตอร์ GradAccum Debug.Options คุณลักษณะเพิ่มเติมสำหรับ LoadTPUEmbeddingMomentumParametersGradAccumDebug
โหลดTPUEmbeddingProximalAdagradParameters โหลดพารามิเตอร์การฝัง Adagrad ใกล้เคียง
โหลดTPUEmbeddingProximalAdagradParameters.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingProximalAdagradParameters
โหลด TPUmbeddingProximalAdagradParametersGradAccumDebug โหลดพารามิเตอร์การฝัง Adagrad ใกล้เคียงพร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
โหลด TPUmbeddingProximalAdagradParametersGradAccumDebug.Options คุณลักษณะเพิ่มเติมสำหรับ LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug
โหลดพารามิเตอร์ TPUUEmbeddingProximalYogi
โหลดTPUEmbeddingProximalYogiParameters.Options คุณลักษณะเพิ่มเติมสำหรับ LoadTPUEmbeddingProximalYogiParameters
โหลดTPUEการฝังProximalYogiParametersGradAccumDebug
โหลด TPU การฝัง ProximalYogiParametersGradAccumDebug.Options คุณลักษณะทางเลือกสำหรับ LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
โหลดTPUEmbeddingRMSPropParameters โหลดพารามิเตอร์การฝัง RMSProp
โหลดTPUEmbeddingRMSPropParameters.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingRMSPropParameters
โหลดTPUEmbeddingRMSPropParametersGradAccumDebug โหลดพารามิเตอร์การฝัง RMSProp ด้วยการสนับสนุนการดีบัก
โหลดTPUEmbeddingRMSPropParametersGradAccumDebug.Options แอตทริบิวต์เพิ่มเติมสำหรับ LoadTPUEmbeddingRMSPropParametersGradAccumDebug
โหลดTPUEการฝังพารามิเตอร์ StochasticGradientDescent โหลดพารามิเตอร์การฝัง SGD
โหลด TPU การฝัง StochasticGradientDescentParameters.Options คุณลักษณะทางเลือกสำหรับ LoadTPUEmbeddingStochasticGradientDescentParameters
โหลดTPUการฝังStochasticGradientDescentParametersGradAccumDebug โหลดพารามิเตอร์การฝัง SGD
โหลด TPU การฝัง StochasticGradientDescentParametersGradAccumDebug.Options คุณลักษณะทางเลือกสำหรับ LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
ตัวเลือกลำดับ ตัวเลือกแกน TPU Op
OutfeedDequeue <T ขยาย TType > ดึงข้อมูลเทนเซอร์ตัวเดียวจากเอาท์พุตการคำนวณ
OutfeedDequeue ตัวเลือก แอ็ตทริบิวต์ทางเลือกสำหรับ OutfeedDequeue
OutfeedDequeueTuple ดึงค่าหลายค่าจากเอาท์พุตการคำนวณ
OutfeedDequeueTuple ตัวเลือก แอ็ตทริบิวต์ทางเลือกสำหรับ OutfeedDequeueTuple
OutfeedDequeueTupleV2 ดึงค่าหลายค่าจากเอาท์พุตการคำนวณ
OutfeedDequeueV2 <T ขยาย TType > ดึงข้อมูลเทนเซอร์ตัวเดียวจากเอาท์พุตการคำนวณ
OutfeedEnqueue จัดคิวเทนเซอร์บนเอาท์พุตการคำนวณ
OutfeedEnqueueTuple จัดคิวค่า Tensor หลายค่าบนเอาท์พุตการคำนวณ
PartitionedInput <T ขยาย TType > สหกรณ์ที่จัดกลุ่มรายการอินพุตที่แบ่งพาร์ติชันไว้ด้วยกัน
PartitionedInput.Options แอ็ตทริบิวต์ทางเลือกสำหรับ PartitionedInput
PartitionedOutput <T ขยาย TType > สหกรณ์ที่แยกเมตริกซ์เทนเซอร์ที่จะแบ่งส่วนโดย XLA ไปยังรายการพาร์ติชัน

เอาต์พุตนอกการคำนวณ XLA

PartitionedOutput.Options แอ็ตทริบิวต์ทางเลือกสำหรับ PartitionedOutput
พรีลิเนียร์ op ที่ทำให้ค่าเทนเซอร์หนึ่งค่าเป็นเส้นตรงไปจนถึงเทนเซอร์ตัวแปรทึบแสง
ปรับเชิงเส้นตัวเลือก แอ็ตทริบิวต์ทางเลือกสำหรับ Prelinearize
พรีลิเนียร์ไรซ์Tuple op ที่ทำให้ค่าเทนเซอร์หลายค่าเป็นเส้นตรงให้เป็นเทนเซอร์ตัวแปรทึบแสง
PrelinearizeTuple.Options แอ็ตทริบิวต์ทางเลือกสำหรับ PrelinearizeTuple
RecvTPUEmbeddingActivations ปฏิบัติการที่ได้รับการเปิดใช้งานการฝังบน TPU
ReplicatedInput <T ขยาย TType > เชื่อมต่ออินพุต N กับการคำนวณ TPU ที่จำลองแบบ N-way
ReplicatedInput.Options แอ็ตทริบิวต์ทางเลือกสำหรับ ReplicatedInput
ReplicatedOutput <T ขยาย TType > เชื่อมต่อเอาต์พุต N จากการคำนวณ TPU ที่จำลองแบบ N-way
จำลองข้อมูลเมตา ข้อมูลเมตาที่ระบุว่าควรจำลองการคำนวณ TPU อย่างไร
ReplicateMetadata.Options แอ็ตทริบิวต์ทางเลือกสำหรับ ReplicateMetadata
ดึงข้อมูล TPUembedAdadeltaParameters ดึงข้อมูลพารามิเตอร์การฝัง Adadelta
เรียกข้อมูล TPUUEmbeddingAdadeltaParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingAdadeltaParameters
ดึงข้อมูล TPU การฝัง AdadeltaParametersGradAccumDebug ดึงข้อมูลพารามิเตอร์การฝัง Adadelta ด้วยการสนับสนุนการดีบัก
ดึงข้อมูล TPU การฝัง AdadeltaParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug
ดึงข้อมูล TPUembedAdagradParameters ดึงข้อมูลพารามิเตอร์การฝัง Adagrad
เรียกข้อมูล TPUUEmbeddingAdagradParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingAdagradParameters
ดึงข้อมูล TPU การฝัง AdagradParametersGradAccumDebug เรียกข้อมูลพารามิเตอร์การฝัง Adagrad ด้วยการสนับสนุนการดีบัก
ดึงข้อมูล TPUEmbeddingAdagradParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingAdagradParametersGradAccumDebug
ดึงTPUEmbeddingADAMพารามิเตอร์ ดึงข้อมูลพารามิเตอร์การฝัง ADAM
เรียกข้อมูล TPUUEmbedding ADAMParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingADAMParameters
ดึงข้อมูล TPU การฝัง ADAMParametersGradAccumDebug ดึงพารามิเตอร์การฝัง ADAM พร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
ดึงข้อมูล TPU การฝัง ADAMParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingADAMParametersGradAccumDebug
ดึงTPUEmbeddingCenteredRMSPropParameters ดึงข้อมูลพารามิเตอร์การฝัง RMSProp ที่กึ่งกลาง
เรียกข้อมูล TPUmbeddingCenteredRMSPropParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingCenteredRMSPropParameters
ดึงTPUEmbeddingFTRLParameters รับพารามิเตอร์การฝัง FTRL
เรียกข้อมูล TPUUEmbeddingFTRLParameters.Options แอ็ตทริบิวต์เผื่อเลือกสำหรับ RetrieveTPUEmbeddingFTRLParameters
ดึงข้อมูล TPU การฝัง FTRLParametersGradAccumDebug รับพารามิเตอร์การฝัง FTRL ด้วยการสนับสนุนการแก้ไขข้อบกพร่อง
เรียกข้อมูล TPUembedFTRLParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingFTRLParametersGradAccumDebug
ดึงข้อมูล TPUembedMDLAdagradLightParameters เรียกข้อมูลพารามิเตอร์การฝัง MDL Adagrad Light
ดึงข้อมูล TPUmbeddingMDLAdagradLightParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingMDLAdagradLightParameters
ดึงพารามิเตอร์การฝัง TPUUE โมเมนตัม รับพารามิเตอร์การฝังโมเมนตัม
เรียกข้อมูล TPUUEmbeddingMomentumParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingMomentumParameters
ดึงข้อมูล TPU การฝังโมเมนตัมพารามิเตอร์ GradAccum Debug ดึงข้อมูลพารามิเตอร์การฝังโมเมนตัมด้วยการสนับสนุนการแก้ไขข้อบกพร่อง
ดึงข้อมูล TPU การฝังโมเมนตัมพารามิเตอร์ GradAccum Debug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
เรียกข้อมูลพารามิเตอร์ TPUUEmbeddingProximalAdagrad ดึงข้อมูลพารามิเตอร์การฝัง Adagrad ที่ใกล้เคียง
เรียกข้อมูล TPUUEmbeddingProximalAdagradParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingProximalAdagradParameters
ดึงข้อมูล TPUUEmbeddingProximalAdagradParametersGradAccumDebug ดึงพารามิเตอร์การฝัง Adagrad ใกล้เคียงพร้อมการสนับสนุนการแก้ไขจุดบกพร่อง
เรียกข้อมูล TPUUEmbeddingProximalAdagradParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug
ดึงข้อมูล TPUUEmbeddingProximalYogiParameters
เรียกข้อมูล TPUUEmbeddingProximalYogiParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingProximalYogiParameters
ดึงข้อมูล TPUEmbeddingProximalYogiParametersGradAccumDebug
เรียกข้อมูล TPUUEmbeddingProximalYogiParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
เรียกข้อมูล TPUUEmbeddingRMSPropParameters เรียกพารามิเตอร์การฝัง RMSProp
เรียกข้อมูล TPUUEmbeddingRMSPropParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingRMSPropParameters
เรียกข้อมูล TPUUEmbeddingRMSPropParametersGradAccumDebug ดึงข้อมูลพารามิเตอร์การฝัง RMSProp ด้วยการสนับสนุนการดีบัก
เรียกข้อมูล TPUUEmbeddingRMSPropParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug
ดึงข้อมูล TPU การฝัง StochasticGradientDescentParameters รับพารามิเตอร์การฝัง SGD
เรียกข้อมูล TPUEmbeddingStochasticGradientDescentParameters.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingStochasticGradientDescentParameters
ดึงข้อมูล TPUE การฝัง StochasticGradientDescentParametersGradAccumDebug รับพารามิเตอร์การฝัง SGD พร้อมการสนับสนุนการแก้ไขข้อบกพร่อง
ดึงข้อมูล TPU การฝัง StochasticGradientDescentParametersGradAccumDebug.Options แอ็ตทริบิวต์ทางเลือกสำหรับ RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
SendTPUการฝังการไล่ระดับสี ดำเนินการอัปเดตการไล่ระดับสีของตารางที่ฝัง
ปิดระบบกระจายTPU ปิดระบบ TPU แบบกระจายที่ทำงานอยู่
ผลการรวบรวม TPU ส่งกลับผลลัพธ์ของการคอมไพล์ TPU
การเปิดใช้งานการฝัง TPU การดำเนินการที่ทำให้เกิดความแตกต่างของ TPU Embeddings
TPUReplicatedInput <T ขยาย TType > เชื่อมต่ออินพุต N กับการคำนวณ TPU ที่จำลองแบบ N-way
TPUReplicatedInput ตัวเลือก แอตทริบิวต์ทางเลือกสำหรับ TPUReplicatedInput
TPUReplicatedOutput <T ขยาย TType > เชื่อมต่อเอาต์พุต N จากการคำนวณ TPU ที่จำลองแบบ N-way
TPUReplicateMetadata ข้อมูลเมตาที่ระบุว่าควรจำลองการคำนวณ TPU อย่างไร
TPUReplicateMetadata ตัวเลือก แอตทริบิวต์ที่ไม่บังคับสำหรับ TPUReplicateMetadata
คนงานการเต้นของหัวใจ คนงาน heartbeat สหกรณ์