Logo
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
Next

Revision 3bc58d4a.

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