MCPcopy Create free account

hub / github.com/KarelPeeters/Kyanite / functions

Functions1,306 in github.com/KarelPeeters/Kyanite

⨍ Functions 1,306◇ Types & classes 523
↓ 2 callersFunctionhtrunc
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1990
↓ 2 callersMethodidentity
(self, dtype: DType)
kn-graph/src/graph.rs:1746
↓ 2 callersMethodidentity_t
(self)
kn-graph/src/graph.rs:1756
↓ 2 callersFunctioninfer_batch_size
Infer what the batch size should be for the given input shapes. Returns: `Ok(Some(batch_size))` if the batch size can be inferred `Ok(None)` if any b
kn-graph/src/shape.rs:360
↓ 2 callersMethodinfer_batch_size
(&self, inputs: Vec<&PyUntypedArray>)
kn-python/src/lib.rs:48
↓ 2 callersMethodinsert_mapping
(&mut self, old: Value, new: Value)
kn-graph/src/optimizer/core.rs:55
↓ 2 callersMethodis_const
(&self, value: Value)
kn-graph/src/graph.rs:425
↓ 2 callersMethodkeeps_spatial_shape
(&self)
kn-graph/src/graph.rs:341
↓ 2 callersFunctionlatent_to_image
(latent: &Tensor<f32>)
kn-runtime/examples/stable_diffusion.rs:225
↓ 2 callersMethodleftover
(&mut self)
kn-graph/src/onnx/inputs.rs:77
↓ 2 callersMethodlinear
(&mut self, input: Value, weight: Value)
kn-graph/src/graph.rs:1053
↓ 2 callersFunctionload_check_data
Load the check data into `(batch_size, inputs, expected_outputs)`.
kn-cuda-eval/src/tester.rs:305
↓ 2 callersFunctionload_check_values
Load the given values from the buffer while advancing it.
kn-cuda-eval/src/tester.rs:333
↓ 2 callersFunctionload_graph_from_onnx_bytes
Load an [ONNX](https://github.com/onnx/onnx/blob/main/docs/IR.md) file from the given bytes. The file is not allowed to reference external data files
kn-graph/src/onnx/mod.rs:39
↓ 2 callersFunctionmake_float2
kn-cuda-sys/doc_headers/cuda_include/vector_functions.hpp:238
↓ 2 callersFunctionmap_bool
(node: Node<&str>, key: &str, i: i64)
kn-graph/src/onnx/inputs.rs:197
↓ 2 callersMethodmap_t
(self, left: T, right: T)
kn-graph/src/graph.rs:1732
↓ 2 callersMethodmaybe_take_bool
(&mut self, key: &str)
kn-graph/src/onnx/inputs.rs:154
↓ 2 callersMethodmaybe_take_float
(&mut self, key: &str)
kn-graph/src/onnx/inputs.rs:174
↓ 2 callersMethodoffset_bytes
(self, offset_bytes: isize)
kn-cuda-eval/src/planner.rs:1134
↓ 2 callersMethodoperation
(self)
kn-graph/src/graph.rs:1760
↓ 2 callersMethodoptimize
(&self)
kn-python/src/lib.rs:42
↓ 2 callersMethodpad
Pad `input` with the given `padding` along each axis using the given `value`. Each padding is a pair of `(before, after)` values.
kn-graph/src/graph.rs:958
↓ 2 callersMethodplan_scalar_op
(&mut self, operation: &str, operands: Vec<Operand<PlanTensor>>, debug_value: Value)
kn-cuda-eval/src/planner.rs:947
↓ 2 callersMethodprepare
(cls, model, device: str = "cpu")
kn-python/example.py:79
↓ 2 callersMethodproperties
(self)
kn-cuda-sys/src/wrapper/handle.rs:69
↓ 2 callersMethodpush_operand_x
(&mut self, tensor: &PlanTensor)
kn-cuda-eval/src/planner.rs:1037
↓ 2 callersMethodrealize_ptr
(&self, ptr: PlanPtr)
kn-cuda-eval/src/planner.rs:1090
↓ 2 callersMethodrealize_tensor
(&self, tensor: PlanTensor)
kn-cuda-eval/src/planner.rs:1109
↓ 2 callersMethodrepeat_impl
(&mut self, input: Value, axis: usize, count: Size, inner: bool)
kn-graph/src/graph.rs:829
↓ 2 callersMethodrepeat_unary
(&mut self, input: Value, axis: usize, count: Size)
kn-graph/src/graph.rs:681
↓ 2 callersFunctionreversed
(mut v: Vec<T>)
kn-graph/src/optimizer/affine.rs:406
↓ 2 callersMethodrun
(&self, handle: &CudnnHandle)
kn-cuda-sys/src/wrapper/group.rs:35
↓ 2 callersMethodrun
(&mut self)
kn-cuda-eval/tests/root/elementwise.rs:142
↓ 2 callersMethodrun_async
Run the steps in this executor. Does no explicit before/after synchronization, so ensure inputs are written and synchronize before reading outputs.
kn-cuda-eval/src/executor.rs:199
↓ 2 callersFunctionrun_conv_bias_res_activation
Runs `output = act(conv(input, filter) + res + bias)`. `res` can be 0, equal to the output or a separate tensor. `input` must be different from both
kn-cuda-sys/src/wrapper/operation.rs:162
↓ 2 callersFunctionsimple
()
kn-cuda-eval/tests/root/onnx.rs:4
↓ 2 callersFunctionstr_to_tokens
(s: &str)
kn-runtime/examples/stable_diffusion.rs:205
↓ 2 callersMethodstream
(&self)
kn-cuda-sys/src/wrapper/handle.rs:210
↓ 2 callersMethodsub
(self, rhs: Self)
kn-graph/src/onnx/typed_value.rs:241
↓ 2 callersMethodtake_all_variadic
(&mut self)
kn-graph/src/onnx/inputs.rs:63
↓ 2 callersMethodtake_new_values
Return all newly crated values since the last call to `take_new_values`.
kn-graph/src/graph.rs:530
↓ 2 callersFunctiontest_elementwise
(op: impl Fn(f32) -> f32, graph_op: impl Fn(&mut Graph, Value) -> Value)
kn-cuda-eval/tests/root/runner.rs:87
↓ 2 callersFunctiontest_restride
(f: impl FnOnce(&CudnnHandle, &TensorDescriptor, &DevicePtr, &TensorDescriptor, &DevicePtr))
kn-cuda-sys/tests/restride.rs:9
↓ 2 callersMethodtime_elapsed_since
Return the elapsed time since `start` _in seconds_.
kn-cuda-sys/src/wrapper/event.rs:19
↓ 2 callersMethodto_size
(self)
kn-graph/src/onnx/typed_value.rs:152
↓ 2 callersFunctiontokens_to_tensor
(tokens: &[u32])
kn-runtime/examples/stable_diffusion.rs:209
↓ 2 callersFunctiontry_run_cpu_operation
( info: &ValueInfo, mut map: impl FnMut(Value) -> OperationResult, input: impl Fn(usize) -> Operat
kn-graph/src/cpu.rs:118
↓ 2 callersMethodtry_visit_fused_conv
(&mut self, value: Value)
kn-cuda-eval/src/planner.rs:675
↓ 2 callersMethodunwrap_2
(&self)
kn-graph/src/shape.rs:108
↓ 2 callersMethodunwrap_4
(&self)
kn-graph/src/shape.rs:122
↓ 2 callersMethodunwrap_f64
(&self)
kn-graph/src/dtype.rs:569
↓ 2 callersMethodunwrap_proto
(self, field: &'static str)
kn-graph/src/onnx/result.rs:87
↓ 2 callersMethodvisit_completely
(&mut self, value: Value)
kn-cuda-eval/src/planner.rs:290
↓ 2 callersMethodvisit_matmul
(&mut self, value: Value, left: Value, right: Value, batch_first: bool)
kn-cuda-eval/src/planner.rs:612
↓ 2 callersMethodwait_for_event
(&self, event: &CudaEvent)
kn-cuda-sys/src/wrapper/handle.rs:159
↓ 2 callersMethodworkspace_size
( &self, handle: &CudnnHandle, algo: cudnnConvolutionFwdAlgo_t, input: &Tensor
kn-cuda-sys/src/wrapper/descriptor.rs:221
↓ 1 callersFunction__bfloat162short_rz
defined(__CUDACC__) || defined(_NVHPC_CUDA) */
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1436
↓ 1 callersFunction__bfloat162ushort_rz
defined(__CUDACC__) || defined(_NVHPC_CUDA) */
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1690
↓ 1 callersFunction__bfloat16_as_ushort
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:2279
↓ 1 callersFunction__cudaCDP2DeviceGetAttribute
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:119
↓ 1 callersFunction__cudaCDP2FuncGetAttributes
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:114
↓ 1 callersFunction__cudaCDP2GetDevice
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:124
↓ 1 callersFunction__cudaCDP2Malloc
__CUDA_INTERNAL_USE_CDP2
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:109
↓ 1 callersFunction__cudaCDP2OccupancyMaxActiveBlocksPerMultiprocessor
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:129
↓ 1 callersFunction__cudaCDP2OccupancyMaxActiveBlocksPerMultiprocessorWithFlags
kn-cuda-sys/doc_headers/cuda_include/cuda_device_runtime_api.h:134
↓ 1 callersFunction__half2short_rz
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1396
↓ 1 callersFunction__half2ushort_rz
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1423
↓ 1 callersFunction__hmax2
* __half2 arithmetic * ************************************************************************
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2313
↓ 1 callersFunction__hmax2_nan
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:3929
↓ 1 callersFunction__hneg2
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:3814
↓ 1 callersFunction__hneg2
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:4032
↓ 1 callersFunction__internal_bfloat162int_rz
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1222
↓ 1 callersFunction__internal_bfloat162uint_rz
defined(__CUDACC__) || defined(_NVHPC_CUDA) */
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1544
↓ 1 callersFunction__internal_device_bfloat162float
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1151
↓ 1 callersFunction__internal_device_bfloat162short_rz
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1422
↓ 1 callersFunction__internal_device_bfloat162ushort_rz
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1676
↓ 1 callersFunction__internal_device_float2_to_half2_rn
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1237
↓ 1 callersFunction__internal_double2float_rn
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:989
↓ 1 callersFunction__internal_float2double
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:999
↓ 1 callersFunction__shfl_down_sync
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2402
↓ 1 callersFunction__shfl_sync
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2388
↓ 1 callersFunction__shfl_up_sync
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2395
↓ 1 callersFunction__shfl_xor_sync
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2409
↓ 1 callersFunction__stwb
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2557
↓ 1 callersMethodalloc
(&mut self, size_bytes: usize)
kn-cuda-sys/src/wrapper/mem/pool.rs:21
↓ 1 callersMethodalloc_tensor_dedicated
(&mut self, shape: ConcreteShape, dtype: DType)
kn-cuda-eval/src/planner.rs:977
↓ 1 callersMethodalloc_tensor_zero
(&mut self, shape: ConcreteShape, dtype: DType, debug_value: Option<Value>)
kn-cuda-eval/src/planner.rs:999
↓ 1 callersMethodapply_fused
(self, settings: OptimizerSettings, graph: &mut Graph, input: Value)
kn-graph/src/optimizer/affine.rs:177
↓ 1 callersFunctionapply_fused_conv
( settings: OptimizerSettings, graph: &mut Graph, input: Value, before: ScaleBias, conv: C
kn-graph/src/optimizer/affine.rs:192
↓ 1 callersFunctionarray_to_tensor
(py: Python, array: &PyUntypedArray)
kn-python/src/lib.rs:80
↓ 1 callersMethodassert_valid
(&self)
kn-graph/src/onnx/typed_value.rs:47
↓ 1 callersMethodbit_cast
(self, to: DType)
kn-graph/src/dtype.rs:355
↓ 1 callersFunctionbuild_operation
(operand_types: &[String], operation: &str)
kn-cuda-eval/src/autokernel/scalar.rs:155
↓ 1 callersFunctioncalculate_reshape_output_shape
(old_shape: &Shape, new_shape_raw: &[SignedSize], allow_zero: bool)
kn-graph/src/onnx/load.rs:1424
↓ 1 callersFunctionchannel_batchnorm
(graph: &mut Graph, input: Value)
kn-cuda-eval/tests/root/graphs.rs:644
↓ 1 callersFunctioncheck_tensor_match
( i: usize, expected_output: &DTensor, output: &DTensor, counts: &mut Counts, first_errors
kn-cuda-eval/src/tester.rs:164
↓ 1 callersFunctioncheck_tensors_match
(expected: &[DTensor], actual: &[DTensor])
kn-cuda-eval/src/tester.rs:140
↓ 1 callersMethodclear_unsafe
(&mut self)
kn-cuda-sys/src/wrapper/mem/pool.rs:48
↓ 1 callersMethodconcat
(mut self, other: &Shape)
kn-graph/src/shape.rs:129
← previousnext →301–400 of 1,306, ranked by callers