org.tensorflow.op.tpu

Các lớp học

AllToAll <T mở rộng TType > Một Op để trao đổi dữ liệu trên các bản sao TPU.
CollectivePermute <T mở rộng TType > Một giải pháp để hoán vị các tensor trên các phiên bản TPU được sao chép.
Biên soạnKết quả Trả về kết quả của quá trình biên dịch TPU.
Biên dịch thành côngKhẳng định Khẳng định rằng quá trình biên dịch đã thành công.
Định cấu hìnhTPU phân tán Thiết lập cấu trúc tập trung cho hệ thống TPU phân tán.
Cấu hìnhDistributedTPU.Options Thuộc tính tùy chọn cho ConfigureDistributedTPU
Định cấu hìnhTPUnhúng Thiết lập TPUEmbedding trong hệ thống TPU phân tán.
CrossReplicaSum <T mở rộng TNumber > Op để tính tổng đầu vào trên các phiên bản TPU được sao chép.
NhúngKích hoạt Một cơ hội cho phép phân biệt các vật liệu nhúng TPU.
EnqueueTPUEnhúngIntegerBatch Một op sắp xếp danh sách các tensor lô đầu vào vào TPUEmbedding.
EnqueueTPUEmbeddingIntegerBatch.Options Thuộc tính tùy chọn cho EnqueueTPUEmbeddingIntegerBatch
EnqueueTPUEembeddingRaggedTensorBatch Giảm bớt việc chuyển mã sử dụng tf.nn.embedding_lookup().
EnqueueTPUEembeddingRaggedTensorBatch.Options Các thuộc tính tùy chọn cho EnqueueTPUEmbeddingRaggedTensorBatch
EnqueueTPUEnhúngSparseBatch Một hoạt động xếp hàng các chỉ số đầu vào TPUEmbedding từ SparseTensor.
EnqueueTPUEembeddingSparseBatch.Options Các thuộc tính tùy chọn cho EnqueueTPUEmbeddingSparseBatch
EnqueueTPUEnhúngSparseTensorBatch Giảm bớt việc chuyển mã sử dụng tf.nn.embedding_lookup_sparse().
EnqueueTPUEembeddingSparseTensorBatch.Options Các thuộc tính tùy chọn cho EnqueueTPUEmbeddingSparseTensorBatch
Hành hình Op tải và thực thi chương trình TPU trên thiết bị TPU.
Thực thiAndUpdateBiến Op thực thi một chương trình với các cập nhật biến tại chỗ tùy chọn.
InfeedDequeue <T mở rộng TType > Một trình giữ chỗ chọn cho một giá trị sẽ được đưa vào tính toán.
InfeedDequeueTuple Tìm nạp nhiều giá trị từ nguồn cấp dữ liệu dưới dạng bộ dữ liệu XLA.
InfeedEnqueue Một op cung cấp một giá trị Tensor duy nhất vào tính toán.
InfeedEnqueue.Options Thuộc tính tùy chọn cho InfeedEnqueue
InfeedEnqueuePrelineizedBuffer Một phương án sắp xếp bộ đệm đã được tuyến tính hóa trước vào tiến trình TPU.
InfeedEnqueuePrelineizedBuffer.Options Các thuộc tính tùy chọn cho InfeedEnqueuePrelinearizedBuffer
InfeedEnqueueTuple Đưa nhiều giá trị Tensor vào tính toán dưới dạng bộ dữ liệu XLA.
InfeedEnqueueTuple.Options Thuộc tính tùy chọn cho InfeedEnqueueTuple
LoadTPUEembeddingAdadeltaThông số Tải các tham số nhúng Adadelta.
LoadTPUEmbeddingAdadeltaParameters.Options Thuộc tính tùy chọn cho LoadTPUEmbeddingAdadeltaParameters
LoadTPUEembeddingAdadeltaParametersGradAccumDebug Tải các tham số Adadelta có hỗ trợ gỡ lỗi.
LoadTPUEembeddingAdadeltaParametersGradAccumDebug.Options Thuộc tính tùy chọn cho LoadTPUEmbeddingAdadeltaParametersGradAccumDebug
LoadTPUEnhúngAdagradThông số Tải các tham số nhúng Adagrad.
LoadTPUEembeddingAdagradParameters.Options Thuộc tính tùy chọn cho LoadTPUEmbeddingAdagradParameters
LoadTPUEnhúngAdagradParametersGradAccumDebug Tải các tham số nhúng Adagrad có hỗ trợ gỡ lỗi.
LoadTPUEembeddingAdagradParametersGradAccumDebug.Options Thuộc tính tùy chọn cho LoadTPUEmbeddingAdagradParametersGradAccumDebug
LoadTPUEnhúngADAMThông số Tải các tham số nhúng ADAM.
LoadTPUEembeddingADAMParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingADAMParameters
LoadTPUEnhúngADAMThông sốGradAccumDebug Tải các tham số nhúng ADAM có hỗ trợ gỡ lỗi.
LoadTPUEembeddingADAMParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingADAMParametersGradAccumDebug
LoadTPUEembeddingCenteredRMSPropThông số Tải các tham số nhúng RMSProp ở giữa.
LoadTPUEmbeddingCenteredRMSPropParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingCenteredRMSPropParameters
LoadTPUEembeddingFTRLThông số Tải các tham số nhúng FTRL.
LoadTPUEembeddingFTRLParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingFTRLParameters
LoadTPUEembeddingFTRLThông sốGradAccumDebug Tải các tham số nhúng FTRL có hỗ trợ gỡ lỗi.
LoadTPUEembeddingFTRLParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingFTRLParametersGradAccumDebug
LoadTPUEembeddingMDLadagradLightParameters Tải các tham số nhúng MDL Adagrad Light.
LoadTPUEembeddingMDLAdagradLightParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingMDLAdagradLightParameters
LoadTPUEembeddingMomentumThông số Tải các tham số nhúng Momentum.
LoadTPUEmbeddingMomentumParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingMomentumParameters
LoadTPUEembeddingMomentumParametersGradAccumDebug Tải các tham số nhúng Momentum có hỗ trợ gỡ lỗi.
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingMomentumParametersGradAccumDebug
LoadTPUEembeddingProximalAdagradThông số Tải các tham số nhúng Adagrad gần nhất.
LoadTPUEmbeddingProximalAdagradParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingProximalAdagradParameters
LoadTPUEembeddingProximalAdagradParametersGradAccumDebug Tải các tham số nhúng Adagrad gần nhất với hỗ trợ gỡ lỗi.
LoadTPUEembeddingProximalAdagradParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug
LoadTPUEembeddingProximalYogiThông số
LoadTPUEmbeddingProximalYogiParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingProximalYogiParameters
LoadTPUEembeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEembeddingRMSPropThông số Tải các tham số nhúng RMSProp.
LoadTPUEmbeddingRMSPropParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingRMSPropParameters
LoadTPUEembeddingRMSPropParametersGradAccumDebug Tải các tham số nhúng RMSProp có hỗ trợ gỡ lỗi.
LoadTPUEembeddingRMSPropParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingRMSPropParametersGradAccumDebug
LoadTPUEembeddingStochasticGradientDescentParameters Tải các tham số nhúng SGD.
LoadTPUEmbeddingStochasticGradientDescentParameters.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingStochasticGradientDescentParameters
LoadTPUEembeddingStochasticGradientDescentParametersGradAccumDebug Tải các tham số nhúng SGD.
LoadTPUEembeddingStochasticGradientDescentParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
Bộ chọn thứ tự Bộ chọn lõi TPU Op.
OutfeedDequeue <T mở rộng TType > Truy xuất một tensor đơn từ nguồn cấp dữ liệu tính toán.
OutfeedDequeue.Options Thuộc tính tùy chọn cho OutfeedDequeue
Nguồn cấp dữ liệu ngoàiDequeueTuple Truy xuất nhiều giá trị từ nguồn cấp dữ liệu tính toán.
OutfeedDequeueTuple.Options Thuộc tính tùy chọn cho OutfeedDequeueTuple
Nguồn cấp dữ liệu raDequeueTupleV2 Truy xuất nhiều giá trị từ nguồn cấp dữ liệu tính toán.
OutfeedDequeueV2 <T mở rộng TType > Truy xuất một tensor đơn từ nguồn cấp dữ liệu tính toán.
OutfeedEnqueue Xếp hàng một Tensor trên nguồn cấp dữ liệu tính toán.
OutfeedEnqueueTuple Xếp hàng đợi nhiều giá trị Tensor trên nguồn cấp dữ liệu tính toán.
Phân vùng đầu vào <T mở rộng TType > Một op nhóm một danh sách các đầu vào được phân vùng lại với nhau.
Phân vùngInput.Options Các thuộc tính tùy chọn cho PartitionedInput
Đầu ra phân vùng <T mở rộng TType > Một op phân tách một tensor được XLA phân chia thành một danh sách được phân vùng

đầu ra bên ngoài tính toán XLA.

Phân vùngOutput.Options Các thuộc tính tùy chọn cho PartitionedOutput
Tuyến tính hóa trước Một op tuyến tính hóa một giá trị Tensor thành một tensor biến thể mờ đục.
Prelineize.Options Các thuộc tính tùy chọn cho Prelinearize
Tuyến tính hóaTuple Một op tuyến tính hóa nhiều giá trị Tensor thành một tensor biến thể mờ đục.
PrelineizeTuple.Options Thuộc tính tùy chọn cho PrelinearizeTuple
RecvTPUEnhúngKích hoạt Một op nhận kích hoạt nhúng trên TPU.
ReplicadInput <T mở rộng TType > Kết nối N đầu vào với tính toán TPU nhân bản N chiều.
Nhân rộngInput.Options Thuộc tính tùy chọn cho ReplicatedInput
ReplicadOutput <T mở rộng TType > Kết nối N đầu ra từ phép tính TPU nhân bản N chiều.
Sao chép siêu dữ liệu Siêu dữ liệu cho biết cách tính toán TPU nên được sao chép.
Sao chépMetadata.Options Các thuộc tính tùy chọn cho ReplicateMetadata
Truy xuấtTPUEnhúngAdadeltaThông số Truy xuất các tham số nhúng Adadelta.
Truy xuấtTPUEembeddingAdadeltaParameters.Options Thuộc tính tùy chọn cho RetrieveTPUEmbeddingAdadeltaParameters
Truy xuấtTPUEnhúngAdadeltaThông sốGradAccumDebug Truy xuất các tham số nhúng Adadelta có hỗ trợ gỡ lỗi.
Truy xuấtTPUEnhúngAdadeltaParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug
Truy xuấtTPUEnhúngAdagradThông số Truy xuất các tham số nhúng Adagrad.
Truy xuấtTPUEnhúngAdagradParameters.Options Thuộc tính tùy chọn cho RetrieveTPUEmbeddingAdagradParameters
Truy xuấtTPUEnhúngAdagradParametersGradAccumDebug Truy xuất các tham số nhúng Adagrad có hỗ trợ gỡ lỗi.
Truy xuấtTPUEnhúngAdagradParametersGradAccumDebug.Options Thuộc tính tùy chọn cho RetrieveTPUEmbeddingAdagradParametersGradAccumDebug
Truy xuấtTPUEnhúngADAMThông số Truy xuất các tham số nhúng ADAM.
Truy xuấtTPUEnhúngADAMParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingADAMParameters
Truy xuấtTPUEnhúngADAMThông sốGradAccumDebug Truy xuất các tham số nhúng ADAM có hỗ trợ gỡ lỗi.
Truy xuấtTPUEnhúngADAMParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingADAMParametersGradAccumDebug
RetrieveTPUEembeddingCenteredRMSPropThông số Truy xuất các tham số nhúng RMSProp ở giữa.
RetrieveTPUEembeddingCenteredRMSPropParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingCenteredRMSPropParameters
Truy xuấtTPUEnhúngFTRLThông số Truy xuất các tham số nhúng FTRL.
Truy xuấtTPUEembeddingFTRLParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingFTRLParameters
Truy xuấtTPUEnhúngFTRLThông sốGradAccumDebug Truy xuất các tham số nhúng FTRL có hỗ trợ gỡ lỗi.
Truy xuấtTPUEnhúngFTRLParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingFTRLParametersGradAccumDebug
Truy xuấtTPUEnhúngMDLAdagradLightParameters Truy xuất các tham số nhúng MDL Adagrad Light.
Truy xuấtTPUEembeddingMDLAdagradLightParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingMDLAdagradLightParameters
Truy xuấtTPUEembeddingMomentumThông số Truy xuất các tham số nhúng Động lượng.
RetrieveTPUEmbeddingMomentumParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingMomentumParameters
Truy xuấtTPUEnhúngĐộng lượngThông sốGradAccumDebug Truy xuất các tham số nhúng Momentum với hỗ trợ gỡ lỗi.
Truy xuấtTPUEembeddingMomentumParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
Truy xuấtTPUEnhúngProximalAdagradThông số Truy xuất các tham số nhúng Adagrad gần nhất.
Truy xuấtTPUEembeddingProximalAdagradParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingProximalAdagradParameters
Truy xuấtTPUEnhúngProximalAdagradThông sốGradAccumDebug Truy xuất các tham số nhúng Adagrad gần nhất với hỗ trợ gỡ lỗi.
Truy xuấtTPUEembeddingProximalAdagradParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug
Truy xuấtTPUEnhúngProximalYogiThông số
Truy xuấtTPUEembeddingProximalYogiParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingProximalYogiParameters
Truy xuấtTPUEnhúngProximalYogiTham sốGradAccumDebug
Truy xuấtTPUEembeddingProximalYogiParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
Truy xuấtTPUEnhúngRMSPropThông số Truy xuất các tham số nhúng RMSProp.
Truy xuấtTPUEembeddingRMSPropParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingRMSPropParameters
Truy xuấtTPUEnhúngRMSPropParametersGradAccumDebug Truy xuất các tham số nhúng RMSProp có hỗ trợ gỡ lỗi.
Truy xuấtTPUEembeddingRMSPropParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug
Truy xuấtTPUEnhúngStochasticGradientDescentParameters Truy xuất các tham số nhúng SGD.
Truy xuấtTPUEembeddingStochasticGradientDescentParameters.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingStochasticGradientDescentParameters
Truy xuấtTPUEnhúngStochasticGradientDescentParametersGradAccumDebug Truy xuất các tham số nhúng SGD có hỗ trợ gỡ lỗi.
Truy xuấtTPUEnhúngStochasticGradientDescentParametersGradAccumDebug.Options Các thuộc tính tùy chọn cho RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
GửiTPUnhúngGradients Thực hiện cập nhật độ dốc của các bảng nhúng.
Tắt máyTPU phân tán Tắt hệ thống TPU phân tán đang chạy.
TPUBiên dịchKết quả Trả về kết quả của quá trình biên dịch TPU.
TPUNhúngKích hoạt Một cơ hội cho phép phân biệt các vật liệu nhúng TPU.
TPUReplicatedInput <T mở rộng TType > Kết nối N đầu vào với tính toán TPU nhân bản N chiều.
TPUReplicatedInput.Options Thuộc tính tùy chọn cho TPUReplicatedInput
TPUReplicatedOutput <T mở rộng TType > Kết nối N đầu ra từ phép tính TPU nhân bản N chiều.
TPUSao chépSiêu dữ liệu Siêu dữ liệu cho biết cách tính toán TPU nên được sao chép.
TPUReplicateMetadata.Options Thuộc tính tùy chọn cho TPUReplicateMetadata
Công NhânNhịp Tim Nhịp tim công nhân op.