nki.isa.engine — AWS Neuron Documentation (original) (raw)
This document is relevant for: Inf2, Trn1, Trn2
nki.isa.engine#
class nki.isa.engine(value)[source]#
Neuron Device engines
Attributes
| tensor | Tensor Engine |
|---|---|
| vector | Vector Engine |
| scalar | Scalar Engine |
| gpsimd | GpSIMD Engine |
| sync | Sync Engine |
| unknown | Unknown Engine |
This document is relevant for: Inf2, Trn1, Trn2