This document is relevant for: Inf2, Trn1

API Reference Guide (tensorflow-neuronx)

Warning

This document is archived. TensorFlow is no longer officially supported by the AWS Neuron SDK. It is provided for reference only. For current framework support, see ML framework support on AWS Neuron SDK.

• 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