Types & classes — github.com/LaurentMazare/xla-rs

CodeHub MCPcopy Create free account

hub / github.com/LaurentMazare/xla-rs / types & classes

Types & classes59 in github.com/LaurentMazare/xla-rs

⨍ Functions 409◇ Types & classes 59

↓ 5 callersClassLiteral

A literal represent a value, typically a multi-dimensional array, stored on the host device.

src/wrappers/literal.rs:7

↓ 2 callersClassCShape

src/wrappers/shape.rs:174

↓ 2 callersClassPjRtClientInternal

src/wrappers/pjrt_client.rs:7

↓ 1 callersClassHloModuleProto

src/wrappers/mod.rs:341

↓ 1 callersClassXlaBuilder

src/wrappers/xla_builder.rs:11

↓ 1 callersClassXlaBuilderInternal

A builder is used to keep track of a computation graph while it's being built.

src/wrappers/xla_builder.rs:8

↓ 1 callersClassXlaComputation

A computation is built from a root [`XlaOp`]. Computations are device independent and can be specialized to a given device through a compilation step.

src/wrappers/mod.rs:292

examples/llama/main.rs:361

InterfaceArrayElement

src/wrappers/mod.rs:142

ClassArrayShape

src/wrappers/shape.rs:5

src/wrappers/mod.rs:271

examples/llama/main.rs:274

examples/nanogpt/main.rs:176

ClassCausalSelfAttention

examples/llama/main.rs:204

ClassCausalSelfAttention

examples/nanogpt/main.rs:113

examples/llama/main.rs:78

EnumElementType

src/wrappers/mod.rs:81

examples/llama/main.rs:105

examples/nanogpt/main.rs:33

src/wrappers/mod.rs:261

InterfaceFromRawBytes

examples/nanogpt/main.rs:213

examples/nanogpt/main.rs:183

examples/nanogpt/main.rs:50

examples/llama/main.rs:121

examples/nanogpt/main.rs:72

examples/llama/main.rs:297

examples/llama/main.rs:175

examples/nanogpt/main.rs:157

examples/llama/var_store.rs:5

InterfaceNativeType

A type implementing the `NativeType` trait can be directly converted to constant ops or literals.

src/wrappers/mod.rs:151

ClassPjRtBuffer

A buffer represents a view on a memory slice hosted on a device.

src/wrappers/pjrt_buffer.rs:6

ClassPjRtClient

src/wrappers/pjrt_client.rs:12

ClassPjRtDevice

A device attached to a [`super::PjRtClient`].

src/wrappers/pjrt_device.rs:5

ClassPjRtLoadedExecutable

src/wrappers/pjrt_loaded_executable.rs:4

EnumPrimitiveType

src/wrappers/mod.rs:33

examples/llama/main.rs:154

src/wrappers/shape.rs:55

examples/llama/sentencepiece.rs:5

examples/nanogpt/tokenizer.rs:265

ClassVarBuilder

examples/llama/var_store.rs:13

examples/llama/var_store.rs:22

examples/nanogpt/var_store.rs:7

src/wrappers/xla_op.rs:11

Class_hlo_module_proto

xla_rs/xla_rs.h:49

xla_rs/xla_rs.h:47

Class_pjrt_buffer

xla_rs/xla_rs.h:42

Class_pjrt_client

xla_rs/xla_rs.h:39

Class_pjrt_device

xla_rs/xla_rs.h:41

Class_pjrt_loaded_executable

xla_rs/xla_rs.h:40

xla_rs/xla_rs.h:46

xla_rs/xla_rs.h:45

Class_xla_builder

xla_rs/xla_rs.h:43

Class_xla_computation

xla_rs/xla_rs.h:48

xla_rs/xla_rs.h:44