Index

A

autotune() (in module intel_extension_for_pytorch.quantization)

C

convert() (in module intel_extension_for_pytorch.quantization) CPUPool (class in intel_extension_for_pytorch.cpu.runtime)

current_device() (in module intel_extension_for_pytorch.xpu) current_stream() (in module intel_extension_for_pytorch.xpu)

D

device (class in intel_extension_for_pytorch.xpu)

device_count() (in module intel_extension_for_pytorch.xpu) device_of (class in intel_extension_for_pytorch.xpu)

E

elapsed_time() (intel_extension_for_pytorch.xpu.Event method) empty_cache() (in module intel_extension_for_pytorch.xpu)

enable_onednn_fusion() (in module intel_extension_for_pytorch) Event (class in intel_extension_for_pytorch.xpu)

F

FrozenBatchNorm2d (class in intel_extension_for_pytorch.nn)

G

get_core_list_of_node_id() (in module intel_extension_for_pytorch.cpu.runtime) get_device_name() (in module intel_extension_for_pytorch.xpu) get_device_properties() (in module intel_extension_for_pytorch.xpu)

get_fp32_math_mode() (in module intel_extension_for_pytorch) get_rng_state() (in module intel_extension_for_pytorch.xpu) get_rng_state_all() (in module intel_extension_for_pytorch.xpu) getDeviceIdListForCard() (in module intel_extension_for_pytorch.xpu)

I

init() (in module intel_extension_for_pytorch.xpu) initial_seed() (in module intel_extension_for_pytorch.xpu) intel_extension_for_pytorch.cpu.runtime module intel_extension_for_pytorch.quantization module

interaction() (in module intel_extension_for_pytorch.nn.functional) is_available() (in module intel_extension_for_pytorch.xpu) is_initialized() (in module intel_extension_for_pytorch.xpu) is_runtime_ext_enabled() (in module intel_extension_for_pytorch.cpu.runtime)

M

manual_seed() (in module intel_extension_for_pytorch.xpu) manual_seed_all() (in module intel_extension_for_pytorch.xpu) max_memory_allocated() (in module intel_extension_for_pytorch.xpu) max_memory_reserved() (in module intel_extension_for_pytorch.xpu) memory_allocated() (in module intel_extension_for_pytorch.xpu) memory_reserved() (in module intel_extension_for_pytorch.xpu) memory_snapshot() (in module intel_extension_for_pytorch.xpu)

memory_stats() (in module intel_extension_for_pytorch.xpu) memory_stats_as_nested_dict() (in module intel_extension_for_pytorch.xpu) memory_summary() (in module intel_extension_for_pytorch.xpu) module intel_extension_for_pytorch.cpu.runtime intel_extension_for_pytorch.quantization MultiStreamModule (class in intel_extension_for_pytorch.cpu.runtime) MultiStreamModuleHint (class in intel_extension_for_pytorch.cpu.runtime)

O

optimize() (in module intel_extension_for_pytorch)

P

pin (class in intel_extension_for_pytorch.cpu.runtime)

prepare() (in module intel_extension_for_pytorch.quantization)

Q

query() (intel_extension_for_pytorch.xpu.Event method)

R

record() (intel_extension_for_pytorch.xpu.Event method) record_event() (intel_extension_for_pytorch.xpu.Stream method)

reset_accumulated_memory_stats() (in module intel_extension_for_pytorch.xpu) reset_peak_memory_stats() (in module intel_extension_for_pytorch.xpu)

S

seed() (in module intel_extension_for_pytorch.xpu) seed_all() (in module intel_extension_for_pytorch.xpu) set_device() (in module intel_extension_for_pytorch.xpu) set_fp32_math_mode() (in module intel_extension_for_pytorch) set_rng_state() (in module intel_extension_for_pytorch.xpu)

set_rng_state_all() (in module intel_extension_for_pytorch.xpu) Stream (class in intel_extension_for_pytorch.xpu) stream() (in module intel_extension_for_pytorch.xpu) synchronize() (in module intel_extension_for_pytorch.xpu) (intel_extension_for_pytorch.xpu.Event method) (intel_extension_for_pytorch.xpu.Stream method)

T

Task (class in intel_extension_for_pytorch.cpu.runtime)

V

verbose (class in intel_extension_for_pytorch)

W

wait() (intel_extension_for_pytorch.xpu.Event method)

wait_event() (intel_extension_for_pytorch.xpu.Stream method) wait_stream() (intel_extension_for_pytorch.xpu.Stream method)

X

xpu::FP32_MATH_MODE (C++ enum) xpu::FP32_MATH_MODE::BF32 (C++ enumerator) xpu::FP32_MATH_MODE::FP32 (C++ enumerator)

xpu::FP32_MATH_MODE::FP32_MATH_MODE_MAX (C++ enumerator) xpu::FP32_MATH_MODE::TF32 (C++ enumerator) xpu::get_queue_from_stream (C++ function) xpu::set_fp32_math_mode (C++ function)