MCPcopy Create free account

hub / github.com/KarelPeeters/Kyanite / functions

Functions1,306 in github.com/KarelPeeters/Kyanite

⨍ Functions 1,306◇ Types & classes 523
↓ 308 callersMethodunwrap
(&self)
kn-cuda-sys/src/wrapper/status.rs:19
↓ 128 callersFunctionNV_IF_ELSE_TARGET
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1312
↓ 127 callersMethodmap
(self, x: DScalar)
kn-graph/src/graph.rs:1649
↓ 115 callersFunctionNV_IF_ELSE_TARGET
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2202
↓ 81 callersFunction__bfloat162float
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1175
↓ 63 callersMethodlen
(&self)
kn-graph/src/dtype.rs:541
↓ 61 callersFunction__half2float
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1300
↓ 59 callersMethodto_owned
(self)
kn-graph/src/onnx/result.rs:93
↓ 57 callersMethodinput
(&mut self, shape: Shape, dtype: DType)
kn-graph/src/graph.rs:585
↓ 56 callersFunctiontest_all
(graph: &Graph, batch_size: usize, inputs: &[DTensor], expected_outputs: Option<&[DTensor]>)
kn-cuda-eval/tests/root/runner.rs:10
↓ 55 callersMethodinner
(&self)
kn-cuda-sys/src/wrapper/graph.rs:41
↓ 55 callersMethodpush
(&mut self, step: PlanStep, debug_value: Value)
kn-cuda-eval/src/planner.rs:666
↓ 52 callersFunction__hisnan
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:3802
↓ 49 callersMethodoutput
Register an existing value as an output
kn-graph/src/graph.rs:1354
↓ 47 callersMethodrequired
(&mut self, index: usize)
kn-graph/src/onnx/inputs.rs:45
↓ 46 callersMethodpush
(&mut self, shape: Shape, dtype: DType, operation: Operation)
kn-graph/src/graph.rs:535
↓ 44 callersMethodptr
(&self)
kn-cuda-eval/src/offset_tensor.rs:31
↓ 44 callersMethodstrided_shape
(&self)
kn-cuda-eval/src/offset_tensor.rs:35
↓ 44 callersMethodunwrap_value
(&self)
kn-graph/src/onnx/typed_value.rs:53
↓ 41 callersMethodptr
(&self)
kn-cuda-sys/src/wrapper/mem/device.rs:92
↓ 33 callersMethodsize
(&self)
kn-cuda-eval/src/shape.rs:77
↓ 32 callersMethodview
(&mut self, input: Value, new_shape: Shape)
kn-graph/src/graph.rs:618
↓ 29 callersMethodshape
(&self)
kn-cuda-eval/src/shape.rs:53
↓ 28 callersMethodconv
( &mut self, input: Value, filter: Value, stride_y: usize, stride_x: u
kn-graph/src/graph.rs:977
↓ 27 callersMethodbinary
(&mut self, op: BinaryOp, left: Value, right: Value)
kn-graph/src/graph.rs:1282
↓ 25 callersMethodrank
(&self)
kn-graph/src/dtype.rs:537
↓ 24 callersMethodtake
(&mut self, index: usize)
kn-graph/src/onnx/inputs.rs:84
↓ 24 callersMethodunary
(&mut self, op: UnaryOp, mut input: Value)
kn-graph/src/graph.rs:1248
↓ 23 callersMethodalloc
(self, len_bytes: usize)
kn-cuda-sys/src/wrapper/handle.rs:65
↓ 23 callersMethodas_ref
(&self)
kn-cuda-eval/src/step.rs:298
↓ 23 callersFunctionlinspace_vec
(len: usize)
kn-cuda-eval/tests/root/tensor_utils.rs:31
↓ 23 callersMethodstrides
(&self)
kn-cuda-eval/src/shape.rs:57
↓ 22 callersMethodadd
(&mut self, left: Value, right: Value)
kn-graph/src/graph.rs:1227
↓ 21 callersMethodshape_dtype
(&self, value: Value)
kn-graph/src/graph.rs:384
↓ 20 callersMethodunwrap_in_drop
(&self)
kn-cuda-sys/src/wrapper/status.rs:28
↓ 19 callersFunction__float2half
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1141
↓ 19 callersFunction__nv_bfloat162
* Move constructor, available for \p C++11 and later dialects */
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:758
↓ 19 callersFunctioncuCimag
kn-cuda-sys/doc_headers/cuda_include/cuComplex.h:193
↓ 19 callersFunctioncuCimagf
kn-cuda-sys/doc_headers/cuda_include/cuComplex.h:84
↓ 19 callersFunctioncuCreal
kn-cuda-sys/doc_headers/cuda_include/cuComplex.h:188
↓ 19 callersFunctioncuCrealf
kn-cuda-sys/doc_headers/cuda_include/cuComplex.h:79
↓ 19 callersMethodunwrap_f32
(self)
kn-graph/src/dtype.rs:260
↓ 19 callersMethodvisit
(&mut self, value: Value)
kn-cuda-eval/src/planner.rs:299
↓ 18 callersMethodget
(&'a self, key: &str)
kn-graph/src/onnx/store.rs:19
↓ 18 callersMethodinputs
(&self)
kn-graph/src/graph.rs:220
↓ 18 callersMethodshape
(&self, graph: &Graph)
kn-graph/src/onnx/typed_value.rs:107
↓ 17 callersFunction__half2
* Move constructor, available for \p C++11 and later dialects */
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:772
↓ 17 callersMethoddtype
(&self)
kn-cuda-eval/src/offset_tensor.rs:39
↓ 17 callersMethodreshape
(&self, shape: E)
kn-graph/src/dtype.rs:549
↓ 16 callersMethodas_slice
(&self)
kn-cuda-sys/src/wrapper/mem/pinned.rs:48
↓ 16 callersMethodbroadcast
(&mut self, input: Value, new_shape: Shape)
kn-graph/src/graph.rs:645
↓ 16 callersMethodcopy_linear_from_host
(&self, buffer: &[u8])
kn-cuda-sys/src/wrapper/mem/device.rs:101
↓ 16 callersFunctionlinspace_tensor
(shape: I)
kn-cuda-eval/tests/root/tensor_utils.rs:17
↓ 16 callersMethodmap_ptr
(self, f: impl FnOnce(P) -> K)
kn-cuda-eval/src/offset_tensor.rs:47
↓ 16 callersMethodslice
(&self, axis: usize, range: SliceRange)
kn-cuda-eval/src/shape.rs:81
↓ 16 callersMethodunwrap_fixed
(self)
kn-graph/src/onnx/typed_value.rs:168
↓ 15 callersMethodconcat
( &mut self, inputs: Vec<Value>, axis: usize, base_shape: Option<Shape>,
kn-graph/src/graph.rs:901
↓ 14 callersMethodinsert
(&self, axis: usize, size: Size)
kn-graph/src/shape.rs:193
↓ 14 callersMethodoptional
(&mut self, index: usize)
kn-graph/src/onnx/inputs.rs:52
↓ 14 callersMethodoutput_all
Register multiple values as output at once, in order.
kn-graph/src/graph.rs:1359
↓ 14 callersMethodshape
(&self)
kn-graph/src/dtype.rs:533
↓ 14 callersMethodswitch_to
Set the current cuda device to this device. TODO is this enough when there are multiple threads running?
kn-cuda-sys/src/wrapper/handle.rs:61
↓ 13 callersFunction__float2half_rn
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:1158
↓ 13 callersMethodconstant_tensor
(&mut self, tensor: DTensor)
kn-graph/src/graph.rs:593
↓ 13 callersMethodcontains
(&self, key: &str)
kn-graph/src/onnx/store.rs:10
↓ 13 callersMethodreplace
Build a new shape with the shape at `axis` replaced by `replacement`, the rest are kept as-is.
kn-graph/src/shape.rs:139
↓ 13 callersFunctionrng_tensor_f32
(shape: I, rng: &mut impl Rng)
kn-cuda-eval/tests/root/tensor_utils.rs:7
↓ 13 callersMethodslice
(&mut self, input: Value, axis: usize, range: SliceRange)
kn-graph/src/graph.rs:773
↓ 13 callersMethodview
(&self, new_shape: Vec<usize>)
kn-cuda-eval/src/shape.rs:133
↓ 12 callersFunctionabs_axis
(axis: i64, rank: usize)
kn-graph/src/onnx/load.rs:1391
↓ 12 callersMethodalloc_tensor_shared
(&mut self, shape: ConcreteShape, dtype: DType, debug_value: Option<Value>)
kn-cuda-eval/src/planner.rs:993
↓ 12 callersMethodas_const
Try to evaluate `value` as a constant.
kn-graph/src/graph.rs:435
↓ 12 callersMethodderef
(&self)
kn-graph/src/dtype.rs:650
↓ 12 callersMethodinner
(&self)
kn-cuda-sys/src/wrapper/descriptor.rs:92
↓ 12 callersMethodmul
(&mut self, left: Value, right: Value)
kn-graph/src/graph.rs:1237
↓ 12 callersMethodunwrap_fixed
(&self, what: &str)
kn-graph/src/shape.rs:89
↓ 12 callersMethodvalue
(&mut self, x: X, y: Y)
kn-cuda-eval/tests/root/elementwise.rs:124
↓ 11 callersMethodassert_has_axis
(&self, axis: usize)
kn-graph/src/shape.rs:76
↓ 11 callersMethodinner
(self)
kn-cuda-sys/src/wrapper/handle.rs:55
↓ 11 callersMethodoutputs
(&self)
kn-graph/src/graph.rs:404
↓ 11 callersMethodstream
(&self)
kn-cuda-eval/src/step.rs:325
↓ 10 callersFunction__hmin
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2282
↓ 10 callersFunction__low2half
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2084
↓ 10 callersMethodabs
(tensor: Tensor<f32>)
kn-graph/src/visualize.rs:278
↓ 10 callersMethodall
()
kn-cuda-sys/src/wrapper/handle.rs:45
↓ 10 callersMethodbytes
(self)
kn-graph/src/dtype.rs:200
↓ 10 callersMethodgather
(&mut self, input: Value, axis: usize, indices: Value)
kn-graph/src/graph.rs:854
↓ 10 callersMethodpermute
(&self, permutation: &[usize])
kn-cuda-eval/src/shape.rs:217
↓ 10 callersMethodsigned_index
(&self, index: isize)
kn-runtime/examples/stable_diffusion.rs:447
↓ 10 callersMethodspecials
TODO move specials to type itself, while keeping this one too?
kn-graph/src/dtype.rs:134
↓ 9 callersFunction__halves2half2
kn-cuda-sys/doc_headers/cuda_include/cuda_fp16.hpp:2147
↓ 9 callersFunction__high2float
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1184
↓ 9 callersFunction__internal_device_fast_bf16log2
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:3837
↓ 9 callersFunction__low2float
kn-cuda-sys/doc_headers/cuda_include/cuda_bf16.hpp:1179
↓ 9 callersMethodmaybe_take_int
(&mut self, key: &str)
kn-graph/src/onnx/inputs.rs:146
↓ 9 callersFunctionoptimize_graph
Optimize the given graph according to the given settings. Returns a new, fully independent graph.
kn-graph/src/optimizer/mod.rs:25
↓ 9 callersMethodpush
(&mut self, value: T)
kn-cuda-sys/src/wrapper/rtc/args.rs:30
↓ 9 callersMethodremove
(&self, axis: usize)
kn-cuda-eval/src/shape.rs:272
↓ 9 callersMethodsize
(self)
kn-graph/src/dtype.rs:113
↓ 9 callersMethodstream
(&self)
kn-cuda-eval/src/executor.rs:112
next →1–100 of 1,306, ranked by callers