API Reference Guide (tensorflow-neuronx) — AWS Neuron Documentation (original) (raw)

• Repository

• Suggest edit

• Open issue

• .rst

This document is relevant for: Inf2, Trn1

API Reference Guide (tensorflow-neuronx)#

• TensorFlow 2.x (tensorflow-neuronx) Tracing API
• TensorFlow 2.x (tensorflow-neuronx) Auto Multicore Replication (Beta)
• TensorFlow 2.x (tensorflow-neuronx) analyze_model API

This document is relevant for: Inf2, Trn1