lớp cuối cùng công khai XlaSparseDenseMatmul
Phương pháp công cộng
Đầu ra <Float> | kích hoạt () |
tĩnh XlaSparseDenseMatmul | tạo ( Phạm vi phạm vi, Toán hạng <Integer> rowIds, Toán hạng <?> colIds, Giá trị toán hạng <Float>, Độ lệch toán hạng <?>, Bảng nhúng toán hạng <Float>, MaxIdsPerPartition dài, MaxUniqueIdsPerPartition dài, Kích thước đầu vào dài) Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác XlaSparseDenseMatmul mới. |
Đầu ra <Số nguyên> | rowPointers () |
Đầu ra <Số nguyên> | |
Đầu ra <Float> | |
Đầu ra <Số nguyên> |
Phương pháp kế thừa
Phương pháp công cộng
công khai tĩnh XlaSparseDenseMatmul tạo ( Phạm vi phạm vi , Toán hạng <Integer> rowIds, Toán hạng <?> colIds, Giá trị toán hạng <Float>, Độ lệch toán hạng <?>, Bảng nhúng toán hạng <Float>, MaxIdsPerPartition dài, MaxUniqueIdsPerPartition dài, Kích thước đầu vào dài)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác XlaSparseDenseMatmul mới.
Thông số
phạm vi | phạm vi hiện tại |
---|
Trả lại
- một phiên bản mới của XlaSparseDenseMatmul