MCPcopy Create free account

hub / github.com/KarelPeeters/Kyanite / functions

Functions1,306 in github.com/KarelPeeters/Kyanite

⨍ Functions 1,306◇ Types & classes 523
↓ 1 callersFunctionconvolution
(details: ConvDetails, input: ArrayView4<T>, kernel: ArrayView4<T>)
kn-graph/src/cpu.rs:347
↓ 1 callersMethodcopy_from_as_scalar_op
TODO ideally we would decay to memcpy if possible but callers can already do that, this is this fallback!
kn-cuda-eval/src/device_tensor.rs:73
↓ 1 callersMethodcopy_linear_from_host_async
(&self, buffer: &PinnedMem, stream: &CudaStream)
kn-cuda-sys/src/wrapper/mem/device.rs:114
↓ 1 callersMethodcopy_linear_to_host_async
(&self, buffer: &mut PinnedMem, stream: &CudaStream)
kn-cuda-sys/src/wrapper/mem/device.rs:141
↓ 1 callersFunctioncpu_eval_graph_exec
Evaluate the given graph on the CPU, with the given batch size and inputs, returning the full execution state including profiling information. Prefer
kn-graph/src/cpu.rs:32
↓ 1 callersFunctioncpu_flip
(input: &Tensor<T>, axis: usize)
kn-graph/src/cpu.rs:279
↓ 1 callersFunctioncpu_gather
(input: &Tensor<T>, axis: usize, indices: DTensor)
kn-graph/src/cpu.rs:304
↓ 1 callersFunctioncudaEventCreateWithFlags
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:429
↓ 1 callersFunctioncudaEventDestroy
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:454
↓ 1 callersFunctioncudaEventRecord
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:434
↓ 1 callersFunctioncudaFree
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:464
↓ 1 callersFunctioncudaGetDeviceCount
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:399
↓ 1 callersFunctioncudaGetErrorString
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:389
↓ 1 callersFunctioncudaLaunchDeviceV2_ptsz
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:624
↓ 1 callersFunctioncudaLaunchDevice_ptsz
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:619
↓ 1 callersFunctioncudaStreamDestroy
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:414
↓ 1 callersFunctioncudaStreamWaitEvent
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:419
↓ 1 callersMethodcurrent_channels
(&self)
kn-graph/src/optimizer/affine.rs:131
↓ 1 callersMethoddtype
(&self, graph: &Graph)
kn-graph/src/onnx/typed_value.rs:100
↓ 1 callersFunctionerror
()
kn-cuda-sys/build.rs:41
↓ 1 callersFunctioneval_binary_op
(op: BinaryOp, a: SignedSize, b: SignedSize)
kn-graph/src/onnx/load.rs:1477
↓ 1 callersMethodfilter_descriptor
(&self, dtype: cudnnDataType_t)
kn-cuda-eval/src/shape.rs:255
↓ 1 callersMethodflatten
(&mut self, input: Value, start_axis: usize)
kn-graph/src/graph.rs:705
↓ 1 callersMethodflip
Flip the given `axis`.
kn-graph/src/graph.rs:802
↓ 1 callersMethodfloor_div
(self, rhs: Self)
kn-graph/src/onnx/typed_value.rs:157
↓ 1 callersMethodfor_each_ptr
TODO report elide lifetimes issue in rust plugin
kn-cuda-eval/src/step.rs:162
↓ 1 callersMethodget_function_by_name
(&self, name: &str)
kn-cuda-sys/src/wrapper/rtc/core.rs:308
↓ 1 callersMethodget_prev_sample
( &mut self, sample: Tensor<f32>, timestep: usize, timestep_pr
kn-runtime/examples/stable_diffusion.rs:413
↓ 1 callersFunctiongraph_from_onnx_bytes
we use &dyn to avoid duplicate codegen of this large and non-critical function
kn-graph/src/onnx/load.rs:43
↓ 1 callersFunctiongraph_to_dot
Render the given graph as a graphviz string. This makes no assumptions about the environment.
kn-graph/src/dot.rs:35
↓ 1 callersFunctiongraph_to_svg
Render the given graph as an svg file. This assumes that graphviz is installed and available on the path as `dot`.
kn-graph/src/dot.rs:12
↓ 1 callersMethodgrow_affine_group
(&self, builder: &mut AffineGroupBuilder, operation: &Operation)
kn-graph/src/optimizer/affine.rs:46
↓ 1 callersFunctionhas_dense_strides
Whether the given shape covers every value within its data range. This is equivalent to asking whether any possible permutation of the shape with abs
kn-cuda-eval/src/shape.rs:309
↓ 1 callersMethodhas_dense_strides
(&self)
kn-cuda-eval/src/shape.rs:69
↓ 1 callersMethodindex
(&self, axis: usize, index: usize)
kn-cuda-eval/src/offset_tensor.rs:86
↓ 1 callersFunctioninfer_batch_size_dims
Same as [infer_batch_size], except for individual dimensions.
kn-graph/src/shape.rs:368
↓ 1 callersMethodinit
(&mut self)
kn-runtime/examples/stable_diffusion.rs:327
↓ 1 callersMethodinput_shapes
(&self)
kn-graph/src/graph.rs:400
↓ 1 callersMethodis_const_one
(&self, value: Value)
kn-graph/src/graph.rs:485
↓ 1 callersFunctionis_effectively_constant
(graph: &Graph, value: Value)
kn-graph/src/visualize.rs:255
↓ 1 callersFunctionis_entirely
(array: &ArrayBase<S, D>, value: f32)
kn-graph/src/optimizer/affine.rs:348
↓ 1 callersMethodis_float
(self)
kn-graph/src/dtype.rs:121
↓ 1 callersMethodis_hidden
(&self, value: Value)
kn-graph/src/graph.rs:416
↓ 1 callersMethodis_neg
(self)
kn-graph/src/onnx/typed_value.rs:176
↓ 1 callersMethodis_signed
(self)
kn-graph/src/dtype.rs:117
↓ 1 callersMethodis_success
(&self)
kn-cuda-sys/src/wrapper/status.rs:14
↓ 1 callersFunctionlayernorm
Layernorm along the given axis of the tensor.
kn-graph/src/cpu.rs:461
↓ 1 callersFunctionlink_cuda
()
kn-cuda-sys/build.rs:73
↓ 1 callersFunctionlink_cuda_docs_rs
()
kn-cuda-sys/build.rs:118
↓ 1 callersMethodload_external_data
(&mut self, location: &Path, _: usize, _: Option<usize>, _: usize)
kn-graph/src/onnx/external_data.rs:28
↓ 1 callersFunctionload_model_proto
(buf: &[u8])
kn-graph/src/onnx/load.rs:1488
↓ 1 callersFunctionmain
()
kn-python/example.py:97
↓ 1 callersFunctionmake_cuComplex
kn-cuda-sys/doc_headers/cuda_include/cuComplex.h:300
↓ 1 callersFunctionmake_half2
CUDA vector-types compatible vector creation function (note returns __half2, not half2) */
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1559
↓ 1 callersMethodmap_ptrs
(self, mut f: impl FnMut(P) -> K)
kn-cuda-eval/src/step.rs:102
↓ 1 callersMethodmaybe_take_tensor
(&mut self, key: &str)
kn-graph/src/onnx/inputs.rs:182
↓ 1 callersMethodname
(self)
kn-cuda-sys/src/wrapper/handle.rs:93
↓ 1 callersMethodold_input
(&self)
kn-graph/src/optimizer/affine.rs:173
↓ 1 callersMethodoutput_dtype
(self, x: DType)
kn-graph/src/graph.rs:1613
↓ 1 callersMethodoutput_tensors
(self)
kn-graph/src/cpu.rs:526
↓ 1 callersMethodpad_to
(&mut self, alignment: usize)
kn-cuda-sys/src/wrapper/rtc/args.rs:52
↓ 1 callersFunctionparse_device
(device: &str)
kn-python/src/lib.rs:139
↓ 1 callersFunctionpath_is_normal
(path: &Path)
kn-graph/src/onnx/external_data.rs:72
↓ 1 callersFunctionplatform
()
kn-cuda-sys/build.rs:15
↓ 1 callersFunctionprefix_line_numbers
(s: &str)
kn-cuda-sys/src/wrapper/rtc/core.rs:322
↓ 1 callersFunctionprofile_different_batch_sizes
(device: CudaDevice, graph: &Graph)
kn-cuda-eval/src/bin/profile_net.rs:119
↓ 1 callersFunctionprofile_single_batch_size_cpu
(graph: &Graph, batch_size: usize, n: usize)
kn-cuda-eval/src/bin/profile_net.rs:182
↓ 1 callersFunctionprofile_single_batch_size_cudnn
(device: CudaDevice, graph: &Graph, batch_size: usize, n: usize, skip_io: bool)
kn-cuda-eval/src/bin/profile_net.rs:146
↓ 1 callersMethodptr
(&self)
kn-cuda-sys/src/wrapper/mem/pinned.rs:42
↓ 1 callersMethodptr_operands
(&self)
kn-cuda-eval/src/step.rs:98
↓ 1 callersFunctionpull_bias_through_conv
( settings: OptimizerSettings, details: ConvDetails, before: Option<ArcArray1<f32>>, filter: &
kn-graph/src/optimizer/affine.rs:277
↓ 1 callersMethodpush_affine
(&mut self, operation: AffineOperation)
kn-graph/src/optimizer/affine.rs:142
↓ 1 callersMethodpush_int
(&mut self, value: i32)
kn-cuda-sys/src/wrapper/rtc/args.rs:43
↓ 1 callersMethodrank
(&self)
kn-graph/src/shape.rs:72
↓ 1 callersMethodrealize_step
(&self, step: PlanStep)
kn-cuda-eval/src/planner.rs:1113
↓ 1 callersMethodrecord_existing_event
(&self, event: &CudaEvent)
kn-cuda-sys/src/wrapper/handle.rs:155
↓ 1 callersMethodrelu
(&mut self, input: Value)
kn-graph/src/graph.rs:1185
↓ 1 callersFunctionresolve_tensor_type
(ty: &TypeProto, name: &str, index: usize, input_shaper: &InputShaper)
kn-graph/src/onnx/load.rs:1336
↓ 1 callersFunctionrun_activation
Run `output = act(input)`. `input` and `output` are allowed to point to the same memory, in which case the operation happens in-place.
kn-cuda-sys/src/wrapper/operation.rs:133
↓ 1 callersFunctionrun_cpu_const_operation
(info: &ValueInfo, map: impl FnMut(Value) -> OperationResult)
kn-graph/src/cpu.rs:114
↓ 1 callersFunctionrun_cpu_operation
( info: &ValueInfo, map: &IndexMap<Value, CalculatedValue>, inputs: &[DTensor], batch_size: us
kn-graph/src/cpu.rs:99
↓ 1 callersMethodrun_custom
( &self, stream: &CudaStream, tensors: &[DeviceTensor], items_per_thread: u32,
kn-cuda-eval/src/autokernel/scalar.rs:119
↓ 1 callersMethodset_conv
(&mut self, conv: ConvOperation)
kn-graph/src/optimizer/affine.rs:136
↓ 1 callersMethodset_external_data
(&mut self, external: Box<dyn ExternalDataLoader>)
kn-graph/src/onnx/loader.rs:86
↓ 1 callersMethodset_profile
(&mut self, profile: bool)
kn-cuda-eval/src/executor.rs:260
↓ 1 callersMethodset_timesteps
(&mut self, num_inference_steps: usize, offset: usize)
kn-runtime/examples/stable_diffusion.rs:346
↓ 1 callersFunctionshould_show_value
(graph: &Graph, value: Value)
kn-graph/src/visualize.rs:213
↓ 1 callersMethodsingle
(index: usize)
kn-graph/src/graph.rs:1560
↓ 1 callersMethodslice
(&self, axis: usize, range: impl Into<SliceRange>)
kn-cuda-eval/src/offset_tensor.rs:70
↓ 1 callersFunctionsoftmax
Softmax along the given axis of the tensor. Implementation (and more importantly, the generic bounds) based on softmax within the onnxruntime crate
kn-graph/src/cpu.rs:443
↓ 1 callersMethodsource_with_line_numbers
(&self)
kn-cuda-sys/src/wrapper/rtc/core.rs:317
↓ 1 callersFunctionsplit_padding
(i: i64, f: i64, s: i64, d: i64, up: bool)
kn-graph/src/onnx/load.rs:1217
↓ 1 callersMethodstep_plms
( &mut self, mut model_output: Tensor<f32>, mut timestep: usize,
kn-runtime/examples/stable_diffusion.rs:369
↓ 1 callersMethodstore_operand_y
(&mut self, operand: &PlanTensor, y: usize)
kn-cuda-eval/src/planner.rs:1076
↓ 1 callersMethodsub
(&mut self, left: Value, right: Value)
kn-graph/src/graph.rs:1232
↓ 1 callersMethodsync_after_cublas
(&self)
kn-cuda-eval/src/step.rs:335
↓ 1 callersMethodsync_before_cublas
(&self)
kn-cuda-eval/src/step.rs:330
↓ 1 callersMethodtake_string
(&mut self, key: &str)
kn-graph/src/onnx/inputs.rs:142
↓ 1 callersMethodtake_tensor
(&mut self, key: &str)
kn-graph/src/onnx/inputs.rs:188
↓ 1 callersFunctiontensor_to_array
(py: Python<'_>, tensor: DTensor)
kn-python/src/lib.rs:113
← previousnext →401–500 of 1,306, ranked by callers