การดำเนินการที่กำหนดค่าซอฟต์แวร์ TPUEmbedding บนโฮสต์
วิธีการสาธารณะ
เอาท์พุต <สตริง> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
กำหนดค่า TPUmbeddingMemory แบบคงที่ | สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ <String> commonConfig) วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ ConfigureTPUEmbeddingMemory ใหม่ |
เอาท์พุต <สตริง> | หน่วยความจำการกำหนดค่า () การกำหนดค่าหน่วยความจำที่เข้ารหัสด้วยสตริงประกอบด้วยข้อมูลเมตาเกี่ยวกับการจัดสรรหน่วยความจำที่สงวนไว้สำหรับ TPUEmbedding |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะ เอาท์พุท <String> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สาธารณะคง ConfigureTPUEmbeddingMemory สร้าง ( ขอบเขต ขอบเขต ตัวดำเนินการ <String> commonConfig)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ ConfigureTPUEmbeddingMemory ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
การกำหนดค่าทั่วไป | CommonConfiguration ที่เข้ารหัสด้วยสตริงประกอบด้วยข้อมูลเมตาเกี่ยวกับเอาต์พุตตัวแบ่งพาร์ติชัน TPUEmbedding และขนาด HBM (เป็นไบต์) ที่จำเป็นสำหรับการดำเนินการ |
การส่งคืน
- อินสแตนซ์ใหม่ของ ConfigureTPUmbeddingMemory
สาธารณะ เอาท์พุท <String> memoryConfig ()
การกำหนดค่าหน่วยความจำที่เข้ารหัสด้วยสตริงประกอบด้วยข้อมูลเมตาเกี่ยวกับการจัดสรรหน่วยความจำที่สงวนไว้สำหรับ TPUEmbedding