Expand description
TPU-optimized operations
Functionsยง
- attention_
tpu - Transformer attention (optimized for TPU)
- batch_
matmul_ tpu - Batch matrix multiplication (optimized for TPU)
- conv2d_
tpu - Convolution on TPU
- matmul_
tpu - Matrix multiplication on TPU