Functions — github.com/buyukakyuz/rig

hub / github.com/buyukakyuz/rig / functions

Functions938 in github.com/buyukakyuz/rig

⨍ Functions 938◇ Types & classes 217

↓ 1 callersFunctionrun_inference

(args: InferenceArgs)

crates/rig-cli/src/commands/inference.rs:315

↓ 1 callersMethodrun_multi_stage_generation_loop

( &mut self, request_id: RequestId, prompt_tokens: usize, start_time: Instant,

crates/rig-worker/src/node.rs:482

↓ 1 callersFunctionrun_pipeline

(args: PipelineArgs)

crates/rig-cli/src/commands/pipeline.rs:105

↓ 1 callersFunctionrun_pipeline_create

(client: &CliClient, args: CreatePipelineArgs)

crates/rig-cli/src/commands/pipeline.rs:120

↓ 1 callersFunctionrun_pipeline_get

(client: &CliClient, pipeline_id_str: &str)

crates/rig-cli/src/commands/pipeline.rs:178

↓ 1 callersFunctionrun_pipeline_list

(client: &CliClient)

crates/rig-cli/src/commands/pipeline.rs:201

↓ 1 callersMethodrun_single_stage_generation

( &mut self, request: InferenceRequest, initial_activation: Activation, prompt

crates/rig-worker/src/node.rs:587

↓ 1 callersFunctionrun_status

(args: StatusArgs)

crates/rig-cli/src/commands/status.rs:26

↓ 1 callersFunctionrun_warmup

(partition: &dyn Partition)

crates/rig-worker/src/node.rs:820

↓ 1 callersFunctionrun_worker

(args: WorkerArgs, config: &RigConfig)

crates/rig-cli/src/commands/worker.rs:60

↓ 1 callersMethodsafetensor_layer_path

(&self, layer_idx: usize, component: &str)

crates/rig-runtime-candle/src/model/mod.rs:162

↓ 1 callersMethodsafetensor_weight_names

(&self)

crates/rig-runtime-candle/src/model/llama.rs:35

↓ 1 callersMethodsafetensor_weight_names

(&self)

crates/rig-runtime-candle/src/model/mistral.rs:35

↓ 1 callersMethodsend_generation_started

( &mut self, request_id: RequestId, eos_token: u32, prompt_tokens: usize,

crates/rig-worker/src/coordinator_client.rs:248

↓ 1 callersMethodsend_token_sampled

( &mut self, request_id: RequestId, token: u32, )

crates/rig-worker/src/coordinator_client.rs:277

↓ 1 callersMethodset_eos_token

(&mut self, eos_token: u32)

crates/rig-core/src/types/stop.rs:105

↓ 1 callersMethodset_model_info

(&mut self, model_info: ModelInfo)

crates/rig-worker/src/node.rs:65

↓ 1 callersMethodset_stage

(&mut self, stage: PipelineStage<F::Transport, C>)

crates/rig-worker/src/node.rs:269

↓ 1 callersFunctionsetup_logging

(log_level: &str)

crates/rig-cli/src/main.rs:75

↓ 1 callersMethodshutdown

(&self)

crates/rig-cli/src/commands/inference.rs:285

↓ 1 callersMethodshutdown

(&self)

crates/rig-cli/src/commands/demo.rs:257

↓ 1 callersMethodsin

(&self)

crates/rig-runtime-candle/src/cache.rs:131

↓ 1 callersMethodsize_bytes_for_elements

(&self, num_elements: usize)

crates/rig-core/src/types/tensor.rs:18

↓ 1 callersMethodspec

(&self)

crates/rig-worker/src/stage.rs:437

↓ 1 callersMethodstart_generation_session

( &self, request_id: RequestId, params: &GenerationParams, eos_token: u32,

crates/rig-coordinator/src/state.rs:603

↓ 1 callersMethodstart_heartbeat_task

(&self)

crates/rig-worker/src/node.rs:147

↓ 1 callersMethodstart_peer_listener

(&mut self)

crates/rig-worker/src/node.rs:89

↓ 1 callersMethodstart_streaming_session

( &self, request_id: RequestId, token_tx: mpsc::UnboundedSender<String>, compl

crates/rig-coordinator/src/state.rs:500

↓ 1 callersMethodsubmit_request

( &self, pipeline_id: PipelineId, request: InferenceRequest, )

crates/rig-coordinator/src/state.rs:464

↓ 1 callersMethodtake_generation_decision

( &self, request_id: RequestId, )

crates/rig-coordinator/src/state.rs:562

↓ 1 callersFunctiontensor_to_activation

( tensor: Tensor, dtype: DType, metadata: ActivationMetadata, )

crates/rig-runtime-candle/src/utils.rs:75

↓ 1 callersFunctiontest_node_info

(node_id: NodeId)

crates/rig-coordinator/src/monitor.rs:74

↓ 1 callersMethodtime_to_first_token_ms

(&self)

crates/rig-coordinator/src/generation.rs:77

↓ 1 callersFunctionto_candle_sampling

(params: &SamplingParams)

crates/rig-runtime-candle/src/utils.rs:28

↓ 1 callersMethodto_llama3_config

(&self)

crates/rig-runtime-candle/src/gguf_config.rs:19

↓ 1 callersMethodtokens_per_second

(&self)

crates/rig-core/src/types/request.rs:313

↓ 1 callersMethodtotal_tokens

(&self)

crates/rig-core/src/types/request.rs:307

↓ 1 callersMethodtransport

(&self)

crates/rig-message-bincode/src/channel.rs:31

↓ 1 callersMethodvalidate

(&self)

crates/rig-cli/src/config.rs:55

↓ 1 callersMethodvar_builder

(&self)

crates/rig-runtime-candle/src/weights/safetensor.rs:25

↓ 1 callersMethodvocab_size

(&self)

crates/rig-runtime-candle/src/tokenizer.rs:102

↓ 1 callersMethodwait_for_assignment

(&mut self)

crates/rig-worker/src/node.rs:170

↓ 1 callersMethodwait_for_pipeline_ready

(&self)

crates/rig-cli/src/commands/inference.rs:258

↓ 1 callersMethodwait_for_pipeline_ready

(&self)

crates/rig-cli/src/commands/demo.rs:230

↓ 1 callersMethodwith_dtype

(mut self, dtype: DType)

crates/rig-core/src/types/protocol.rs:364

↓ 1 callersMethodwith_generation_params

(mut self, params: GenerationParams)

crates/rig-core/src/types/tensor.rs:152

↓ 1 callersMethodwith_heartbeat_interval

(mut self, interval: Duration)

crates/rig-worker/src/config.rs:39

↓ 1 callersMethodwith_listen_addr

(mut self, addr: SocketAddr)

crates/rig-worker/src/config.rs:33

↓ 1 callersMethodwith_nodelay

(mut self, nodelay: bool)

crates/rig-transport-tcp/src/config.rs:61

↓ 1 callersMethodwith_params

(mut self, params: GenerationParams)

crates/rig-core/src/types/request.rs:230

↓ 1 callersMethodwith_partition

(mut self, partition: Box<dyn Partition>)

crates/rig-worker/src/stage.rs:337

↓ 1 callersMethodwith_priority

(mut self, priority: Priority)

crates/rig-core/src/types/request.rs:186

↓ 1 callersMethodwith_rope_scaling

(mut self, rope_scaling: Option<Llama3RopeConfig>)

crates/rig-runtime-candle/src/gguf_config.rs:186

Methodaccepted

(node_id: NodeId)

crates/rig-core/src/types/protocol.rs:141

Methodactivation

(&self)

crates/rig-runtime-candle/src/model/llama.rs:15

Methodactivation

(&self)

crates/rig-runtime-candle/src/model/qwen.rs:15

Methodactivation

(&self)

crates/rig-runtime-candle/src/model/mistral.rs:15

Methodactive_slots

(&self)

crates/rig-runtime-candle/src/kv_cache.rs:97

Functionaddress_serialization_roundtrip

()

crates/rig-core/src/types/node.rs:329

Functionaddress_tcp

()

crates/rig-core/src/types/node.rs:299

Functionaddress_tcp_from_host_port

()

crates/rig-core/src/types/node.rs:319

Functionaddress_unix

()

crates/rig-core/src/types/node.rs:311

Methodapply_chat_template

( &self, messages: &[crate::ChatMessage], add_generation_prompt: bool, )

crates/rig-core/src/traits/tokenizer.rs:14

Methodapply_rotary_emb

( q: &Tensor, k: &Tensor, index_pos: usize, rope_cache: &RopeCache, )

crates/rig-runtime-candle/src/layers/attention.rs:97

Methodas_unix

(&self)

crates/rig-core/src/types/node.rs:223

Functionassert_send_sync

()

crates/rig-message-bincode/src/channel.rs:175

Functionassert_send_sync

()

crates/rig-message-bincode/src/lib.rs:165

Methodassignment

(&self)

crates/rig-worker/src/stage.rs:91

Functionassignment_num_layers

()

crates/rig-core/src/types/pipeline.rs:298

Methodassistant

(content: impl Into<String>)

crates/rig-core/src/types/request.rs:73

Methodattention_config

(&self)

crates/rig-runtime-candle/src/model/llama.rs:27

Functionbidirectional_communication

()

crates/rig-transport-tcp/tests/integration.rs:204

Methodbind

(addr: &str)

crates/rig-transport-tcp/src/listener.rs:17

Methodbind_addr

(addr: SocketAddr)

crates/rig-transport-tcp/src/listener.rs:21

Methodbind_addr_with_config

( addr: SocketAddr, config: TcpConfig, )

crates/rig-transport-tcp/src/listener.rs:30

Functionbind_and_accept_with_socket_addr

()

crates/rig-transport-tcp/src/listener.rs:101

Functionbind_and_accept_with_trait

()

crates/rig-transport-tcp/src/listener.rs:131

Methodbind_with_config

(addr: &str, config: TcpConfig)

crates/rig-transport-tcp/src/listener.rs:25

Functionbind_with_socket_addr

()

crates/rig-transport-tcp/src/listener.rs:162

Methodbos_token_str

(&self)

crates/rig-runtime-candle/src/config.rs:162

Methodbuild_pipeline_info_internal

( pipeline_id: PipelineId, record: &PipelineRecord, )

crates/rig-coordinator/src/state.rs:765

Functionbuilder_methods_work

()

crates/rig-transport-tcp/src/config.rs:96

Functionbuilder_pattern

()

crates/rig-worker/src/config.rs:96

Functionbuilder_pattern

()

crates/rig-coordinator/src/config.rs:84

Functionbuilder_requires_assignment

()

crates/rig-worker/src/stage.rs:421

Functionbuilder_requires_partition

()

crates/rig-worker/src/stage.rs:415

Functioncache_slot_capacity

()

crates/rig-core/src/types/tensor.rs:310

Functionchannel_is_send_and_sync

()

crates/rig-message-bincode/src/channel.rs:174

Functioncharacteristics_are_correct

()

crates/rig-transport-tcp/src/factory.rs:194

Methodcheck_heartbeats

(&self)

crates/rig-coordinator/src/monitor.rs:54

Methodcodec

(&self)

crates/rig-message-bincode/src/channel.rs:35

Functioncodec_is_send_and_sync

()

crates/rig-message-bincode/src/lib.rs:164

Methodcompute_even_split

( num_layers: usize, num_stages: usize, )

crates/rig-core/src/types/partitioning.rs:70

Methodcompute_memory_balanced

( num_layers: usize, node_memory_bytes: &[u64], )

crates/rig-core/src/types/partitioning.rs:150

Functionconcurrent_operations

()

crates/rig-transport-tcp/tests/integration.rs:290

Methodconfig

(&self)

crates/rig-transport-tcp/src/factory.rs:31

Methodconfig

(&self)

crates/rig-transport-tcp/src/transport.rs:36

Methodconfig

(&self)

crates/rig-transport-tcp/src/listener.rs:55

Methodconfig

(msg: impl Into<String>)

crates/rig-worker/src/error.rs:69

Methodconfig

(&self)

crates/rig-worker/src/node.rs:75

← previous next →301–400 of 938, ranked by callers