MCPcopy Create free account

hub / github.com/LaurentMazare/xla-rs / functions

Functions409 in github.com/LaurentMazare/xla-rs

⨍ Functions 409◇ Types & classes 59
↓ 1 callersFunctionconstant_literal
xla_rs/xla_rs.cc:180
↓ 1 callersFunctionconvert_state_dict
(state_dict: Dict[str, torch.Tensor], dtype: torch.dtype = torch.float16)
examples/llama/convert_checkpoint.py:11
↓ 1 callersFunctionconvert_weights
(llama_ckpt, *, output_npz: Path = Path("llama.npz"), dtype: str = "float16")
examples/llama/convert_checkpoint.py:56
↓ 1 callersMethoddecode
The inverse of the tokenization process, takes as input a list of tokens and returns a string that produces this tokenization.
examples/nanogpt/tokenizer.rs:390
↓ 1 callersMethodelement_type
The element type used by element stored in this literal.
src/wrappers/literal.rs:82
↓ 1 callersMethodencode
Run bpe on the whole string, very very inefficient but should be good enough for prompts. The original string should first be split on whitespace/...
examples/llama/sentencepiece.rs:140
↓ 1 callersMethodencode
The main tokenization entry point, takes as input a string and returns the list of tokens.
examples/nanogpt/tokenizer.rs:379
↓ 1 callersFunctionenv_var_rerun
(name: &str)
build.rs:55
↓ 1 callersFunctionexecute
xla_rs/xla_rs.cc:890
↓ 1 callersFunctionexecute_b
xla_rs/xla_rs.cc:923
↓ 1 callersMethodexecute_b
( &self, args: &[L], )
src/wrappers/pjrt_loaded_executable.rs:54
↓ 1 callersFunctionfirst_error
xla_rs/xla_rs.cc:880
↓ 1 callersMethodfirst_error
This returns `Ok(())` if the graph creation has not generated any error so far. Otherwise the first error is returned.
src/wrappers/xla_builder.rs:36
↓ 1 callersMethodforward
(&self, indexes: &XlaOp)
examples/llama/main.rs:115
↓ 1 callersMethodforward
(&self, indexes: &XlaOp)
examples/nanogpt/main.rs:43
↓ 1 callersMethodgather
( &self, start_indices: &XlaOp, offset_dims: &[i64], collapsed_slice_dims: &[i
src/wrappers/xla_op.rs:465
↓ 1 callersFunctionget_current_status
xla_rs/xla_rs.cc:885
↓ 1 callersFunctionget_dimensions
xla_rs/xla_rs.cc:855
↓ 1 callersFunctionget_dimensions_size
xla_rs/xla_rs.cc:849
↓ 1 callersMethodget_dims
The dimension sizes associated with this op.
src/wrappers/xla_builder.rs:218
↓ 1 callersFunctionget_element_type
xla_rs/xla_rs.cc:842
↓ 1 callersMethodget_primitive_type
The element type associated with this op.
src/wrappers/xla_builder.rs:227
↓ 1 callersFunctionget_shape
xla_rs/xla_rs.cc:836
↓ 1 callersFunctiongpt_computation
(vs: VarStore, bsize: i64)
examples/nanogpt/main.rs:252
↓ 1 callersFunctionhlo_module_proto_free
xla_rs/xla_rs.cc:1081
↓ 1 callersFunctionhlo_module_proto_parse_and_return_unverified_module
xla_rs/xla_rs.cc:1039
↓ 1 callersFunctionhlo_module_proto_parse_proto
xla_rs/xla_rs.cc:1047
↓ 1 callersMethodid
The device unique identifier.
src/wrappers/pjrt_device.rs:12
↓ 1 callersFunctioninfeed
xla_rs/xla_rs.cc:271
↓ 1 callersMethodinternal_error
An error node, using the 'internal error' error type.
src/wrappers/xla_builder.rs:179
↓ 1 callersMethodinto_store
(self)
examples/llama/var_store.rs:67
↓ 1 callersMethodlayer_norm
Layer normalization, this normalizes values on the target dimension to be of zero mean and standard deviation one, and then scales the result by `scal
src/wrappers/xla_op.rs:582
↓ 1 callersFunctionliteral_clone
xla_rs/xla_rs.cc:968
↓ 1 callersFunctionliteral_convert
xla_rs/xla_rs.cc:980
↓ 1 callersFunctionliteral_copy_from
xla_rs/xla_rs.cc:1007
↓ 1 callersFunctionliteral_copy_to
xla_rs/xla_rs.cc:1003
↓ 1 callersFunctionliteral_create_from_shape
xla_rs/xla_rs.cc:947
↓ 1 callersFunctionliteral_create_from_shape_and_data
xla_rs/xla_rs.cc:955
↓ 1 callersFunctionliteral_decompose_tuple
xla_rs/xla_rs.cc:994
↓ 1 callersFunctionliteral_element_count
xla_rs/xla_rs.cc:986
↓ 1 callersFunctionliteral_element_type
xla_rs/xla_rs.cc:1001
↓ 1 callersFunctionliteral_free
xla_rs/xla_rs.cc:1025
↓ 1 callersFunctionliteral_make_tuple_owned
xla_rs/xla_rs.cc:1016
↓ 1 callersFunctionliteral_reshape
xla_rs/xla_rs.cc:972
↓ 1 callersFunctionliteral_shape
xla_rs/xla_rs.cc:990
↓ 1 callersFunctionliteral_size_bytes
xla_rs/xla_rs.cc:988
↓ 1 callersFunctionllama_computation
(args: &Args, bsize: i64)
examples/llama/main.rs:343
↓ 1 callersMethodload_from_npz
( &mut self, path: P, c: &PjRtClient, )
examples/llama/var_store.rs:106
↓ 1 callersMethodlocal_hardware_id
(&self)
src/wrappers/pjrt_device.rs:20
↓ 1 callersFunctionmake_shape_array
xla_rs/xla_rs.cc:244
↓ 1 callersFunctionmake_shape_tuple
xla_rs/xla_rs.cc:248
↓ 1 callersFunctionmake_shared_lib
(os: OS, xla_dir: P)
build.rs:26
↓ 1 callersFunctionmasked_fill
(on_false: &XlaOp, mask: &XlaOp, on_true: T)
examples/llama/main.rs:197
↓ 1 callersFunctionmasked_fill
(on_false: &XlaOp, mask: &XlaOp, on_true: T)
examples/nanogpt/main.rs:106
↓ 1 callersMethodmax_value
A scalar node with the maximum value for the associated type.
src/wrappers/xla_builder.rs:152
↓ 1 callersMethodmaybe_keep_dims
(&self, res: XlaOp, dims_to_keep: &[i64], keep_dims: bool)
src/wrappers/xla_op.rs:513
↓ 1 callersMethodmin_value
A scalar node with the minimum value for the associated type.
src/wrappers/xla_builder.rs:146
↓ 1 callersFunctionnew_gelu
(x: &XlaOp)
examples/nanogpt/main.rs:24
↓ 1 callersMethodnormalize_indexes
(&self, indexes: &[i64])
src/wrappers/xla_op.rs:307
↓ 1 callersFunctionop_broadcast
xla_rs/xla_rs.cc:604
↓ 1 callersFunctionop_broadcast_in_dim
xla_rs/xla_rs.cc:610
↓ 1 callersFunctionop_clamp
xla_rs/xla_rs.cc:632
↓ 1 callersFunctionop_clone
xla_rs/xla_rs.cc:579
↓ 1 callersFunctionop_collapse
xla_rs/xla_rs.cc:620
↓ 1 callersFunctionop_concat_in_dim
xla_rs/xla_rs.cc:669
↓ 1 callersFunctionop_conditional
xla_rs/xla_rs.cc:807
↓ 1 callersFunctionop_convert_element_type
xla_rs/xla_rs.cc:722
↓ 1 callersFunctionop_dimensions_size
xla_rs/xla_rs.cc:728
↓ 1 callersFunctionop_dot_general
xla_rs/xla_rs.cc:363
↓ 1 callersFunctionop_einsum1
xla_rs/xla_rs.cc:561
↓ 1 callersFunctionop_einsum2
xla_rs/xla_rs.cc:567
↓ 1 callersFunctionop_gather
xla_rs/xla_rs.cc:697
↓ 1 callersFunctionop_get_tuple_element
xla_rs/xla_rs.cc:691
↓ 1 callersFunctionop_internal_error
xla_rs/xla_rs.cc:743
↓ 1 callersFunctionop_invalid_argument_error
xla_rs/xla_rs.cc:755
↓ 1 callersFunctionop_iota
xla_rs/xla_rs.cc:791
↓ 1 callersFunctionop_iota1
xla_rs/xla_rs.cc:785
↓ 1 callersFunctionop_max_value
xla_rs/xla_rs.cc:779
↓ 1 callersFunctionop_min_value
xla_rs/xla_rs.cc:773
↓ 1 callersFunctionop_one
xla_rs/xla_rs.cc:767
↓ 1 callersFunctionop_reduce
xla_rs/xla_rs.cc:734
↓ 1 callersFunctionop_reshape
xla_rs/xla_rs.cc:598
↓ 1 callersFunctionop_rng_normal
xla_rs/xla_rs.cc:653
↓ 1 callersFunctionop_rng_uniform
xla_rs/xla_rs.cc:644
↓ 1 callersFunctionop_select
xla_rs/xla_rs.cc:638
↓ 1 callersFunctionop_slice_in_dim
xla_rs/xla_rs.cc:662
↓ 1 callersFunctionop_transpose
xla_rs/xla_rs.cc:626
↓ 1 callersFunctionop_tuple
xla_rs/xla_rs.cc:681
↓ 1 callersFunctionop_unknown_error
xla_rs/xla_rs.cc:749
↓ 1 callersFunctionop_while
xla_rs/xla_rs.cc:800
↓ 1 callersFunctionop_zero
xla_rs/xla_rs.cc:761
↓ 1 callersFunctionoutfeed
xla_rs/xla_rs.cc:279
↓ 1 callersFunctionparameter
xla_rs/xla_rs.cc:256
↓ 1 callersFunctionparameter_s
xla_rs/xla_rs.cc:264
↓ 1 callersFunctionpjrt_buffer_copy_raw_to_host_sync
xla_rs/xla_rs.cc:136
↓ 1 callersFunctionpjrt_buffer_copy_to_device
xla_rs/xla_rs.cc:129
↓ 1 callersFunctionpjrt_buffer_free
xla_rs/xla_rs.cc:142
↓ 1 callersFunctionpjrt_buffer_from_host_literal
xla_rs/xla_rs.cc:109
↓ 1 callersFunctionpjrt_buffer_on_device_shape
xla_rs/xla_rs.cc:125
↓ 1 callersFunctionpjrt_buffer_to_literal_sync
xla_rs/xla_rs.cc:118
← previousnext →101–200 of 409, ranked by callers