Index

A

abs (C++ function) abs_out (C++ function)

accessor (C++ function), [1] add (C++ function), [1] add_out (C++ function), [1]

B

benchmark() built-in function BF16 bitwise_and (C++ function), [1], [2] bitwise_and_out (C++ function), [1], [2] bitwise_not (C++ function) bitwise_not_out (C++ function) bitwise_or (C++ function), [1], [2] bitwise_or_out (C++ function), [1], [2] built-in function benchmark() compile() get_reports() model_index.append() model_index.copy() model_index.create() model_index.filter() model_index.load() model_index.move() model_index.save() print_reports() torch.neuron.DataParallel() torch.neuron.DataParallel.disable_dynamic_batching() torch_neuron.trace() torch_neuronx.analyze() torch_neuronx.dynamic_batch() torch_neuronx.experimental.multicore_context() torch_neuronx.experimental.neuron_cores_context() torch_neuronx.experimental.profiler.profile() torch_neuronx.experimental.profiler.profile.start() torch_neuronx.experimental.set_multicore() torch_neuronx.experimental.set_neuron_cores() torch_neuronx.trace() write_csv() write_json()

C

CCE ceil (C++ function) ceil_out (C++ function) cFP8 clamp (C++ function) clamp_out (C++ function)

close (C++ function), [1] Collective Communication Engine compile() built-in function cos (C++ function) cos_out (C++ function) CustomOps

D

div (C++ function), [1] div_out (C++ function), [1]

DP DPr

E

empty (C++ function) exp (C++ function)

exp_out (C++ function) eye (C++ function)

F

fill_ (C++ function) FLOAT32_TO_FLOAT16 (torch_neuron.Optimization attribute) floor (C++ function)

floor_out (C++ function) FP16 FP32 full (C++ function)

G

get_accessor_coherence_policy (C++ function) get_reports() built-in function

GPSIMD Engine

I

Inf1	Inferentia

L

log (C++ function) log10 (C++ function) log10_out (C++ function)

log2 (C++ function) log2_out (C++ function) log_out (C++ function)

M

model_index.append() built-in function model_index.copy() built-in function model_index.create() built-in function model_index.filter() built-in function model_index.load() built-in function

model_index.move() built-in function model_index.save() built-in function module placement mul (C++ function), [1] mul_out (C++ function), [1]

N

NC ND Neuron Device neuron-cc neuron-cc command line option, [1], [2] neuron-cc command line option neuron-cc, [1], [2] neuron-ls neuron-ls command line option neuron-ls command line option neuron-ls neuron-monitor neuron-monitor command line option neuron-monitor command line option neuron-monitor NeuronCore, [1] NeuronCore-v1 NeuronCore-v2 NeuronDevice NeuronLink NeuronLink-v1 NeuronLink-v2 neuronx-cc neuronx-cc command line option, [1], [2] neuronx-cc command line option neuronx-cc, [1], [2] nrt_add_tensor_to_tensor_set (C function)

nrt_allocate_tensor_set (C function) nrt_close (C function) nrt_destroy_tensor_set (C function) nrt_execute (C function) nrt_execute_repeat (C function) nrt_free_model_tensor_info (C function) nrt_get_model_instance_count (C function) nrt_get_model_nc_count (C function) nrt_get_model_tensor_info (C function) nrt_get_tensor_from_tensor_set (C function) nrt_get_total_nc_count (C function) nrt_get_version (C function) nrt_get_visible_nc_count (C function) nrt_init (C function) nrt_load (C function) nrt_load_collectives (C function) nrt_profile_start (C function) nrt_profile_stop (C function) nrt_tensor_allocate (C function) nrt_tensor_allocate_empty (C function) nrt_tensor_allocate_slice (C function) nrt_tensor_attach_buffer (C function) nrt_tensor_free (C function) nrt_tensor_get_size (C function) nrt_tensor_get_va (C function) nrt_tensor_read (C function) nrt_tensor_write (C function) nrt_unload (C function)

O

ones (C++ function)

operator= (C++ function), [1]

P

placement module pow (C++ function), [1], [2] pow_out (C++ function), [1], [2]

PP PPr print_reports() built-in function

R

read (C++ function) read_stream_accessor (C++ function)

RNE RT

S

Scalar Engine ScalEng set_accessor_coherence_policy (C++ function) sin (C++ function) sin_out (C++ function)

SR sub (C++ function) sub_out (C++ function), [1] Sync Engine SyncEng

T

tan (C++ function) tan_out (C++ function) tcm_accessor (C++ function), [1] tcm_to_tensor (C++ function) TensEng Tensor Engine tensor_to_tcm (C++ function) TF32 torch.neuron.DataParallel() built-in function torch.neuron.DataParallel.disable_dynamic_batching() built-in function torch::neuron::tcm_free (C++ function) torch::neuron::tcm_malloc (C++ function) torch_neuron.experimental.multicore_context() (in module placement) torch_neuron.experimental.neuron_cores_context() (in module placement) torch_neuron.experimental.set_multicore() (in module placement) torch_neuron.experimental.set_neuron_cores() (in module placement) torch_neuron.Optimization (built-in class) torch_neuron.trace() built-in function

torch_neuronx.analyze() built-in function torch_neuronx.dynamic_batch() built-in function torch_neuronx.experimental.multicore_context() built-in function torch_neuronx.experimental.neuron_cores_context() built-in function torch_neuronx.experimental.profiler.profile() built-in function torch_neuronx.experimental.profiler.profile.start() built-in function torch_neuronx.experimental.set_multicore() built-in function torch_neuronx.experimental.set_neuron_cores() built-in function torch_neuronx.trace() built-in function TP TPr Trainium Trn1

V

VecEng

Vector Engine

W

write (C++ function) write_csv() built-in function

write_json() built-in function write_stream_accessor (C++ function)

Z

zeros (C++ function)