Creating Custom Operations for the IPU
Version: latest
1. Introduction
2. Creating a custom op in PyTorch
2.1. Implementing the custom op
2.2. Make the op available in PyTorch
3. Creating a custom op in TensorFlow
3.1. A custom op that executes on the IPU
3.2. A custom op that executes on the host CPU
4. Creating a custom op in PopART
4.1. Implementing the custom op in PopART
4.2. Make the op available in PopART
5. Examples
6. Trademarks & copyright
Creating Custom Operations for the IPU
»
Creating Custom Operations for the IPU
Edit on GitHub
Creating Custom Operations for the IPU
ΒΆ
1. Introduction
2. Creating a custom op in PyTorch
2.1. Implementing the custom op
2.2. Make the op available in PyTorch
3. Creating a custom op in TensorFlow
3.1. A custom op that executes on the IPU
3.2. A custom op that executes on the host CPU
4. Creating a custom op in PopART
4.1. Implementing the custom op in PopART
4.2. Make the op available in PopART
5. Examples
6. Trademarks & copyright
Read the Docs
v: latest
Versions
latest
2.4.0
2.3.0
2.2.0
2.1.0
1.0.0
Downloads
pdf
html
On Read the Docs
Project Home
Builds