Functions — github.com/ddkang/zkml

hub / github.com/ddkang/zkml / functions

Functions400 in github.com/ddkang/zkml

⨍ Functions 400◇ Types & classes 92

↓ 80 callersMethodshape

(inp: &AssignedTensor<F>, layer_config: &LayerConfig)

src/layers/max_pool_2d.rs:21

↓ 18 callersMethodop_aligned_rows

The caller is required to ensure that the inputs are of the correct length.

src/gadgets/gadget.rs:108

↓ 13 callersMethodload_lookups

(&self, mut layouter: impl Layouter<F>)

src/gadgets/input_lookup.rs:37

↓ 10 callersFunctionconvert_to_u128

(x: &F)

src/gadgets/gadget.rs:81

↓ 9 callersFunctionconvert_to_u64

TODO: refactor

src/gadgets/gadget.rs:66

↓ 9 callersMethodnew_tensor

(self)

backwards/backward.py:27

↓ 7 callersMethodnum_inputs_per_row

(&self)

src/gadgets/max.rs:92

↓ 6 callersMethodforward

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, constants: &HashMap

src/layers/softmax.rs:119

↓ 6 callersMethodnum_inputs_per_row

(&self)

src/gadgets/dot_prod.rs:84

↓ 5 callersMethodnum_cols_per_op

()

src/gadgets/update.rs:30

↓ 4 callersMethodarithmetic_forward

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, constants: &HashMap

src/layers/arithmetic.rs:27

↓ 4 callersFunctionget_public_values

Get the public values

src/utils/helpers.rs:56

↓ 4 callersMethodgradient_tensor_idx

(self, tensor_idx)

backwards/backward.py:40

↓ 4 callersMethodnew_gradient_tensor

(self, tensor_idx)

backwards/backward.py:19

↓ 4 callersFunctionserialize

(data: &Vec<u8>, path: &str)

src/utils/proving_kzg.rs:48

↓ 3 callersMethodforward

IMPORTANT: Assumes input tensors are in order. Output tensors can be in any order.

src/layers/dag.rs:62

↓ 3 callersFunctionget_shape

(interpreter: tf.lite.Interpreter, tensor_idx)

python/converter.py:9

↓ 3 callersFunctionhandle_numpy_or_literal

(inp: Union[np.ndarray, Literal[0]])

python/converter.py:15

↓ 3 callersFunctionload_config_msgpack

(config_path: &str)

src/utils/loader.rs:39

↓ 3 callersMethodnum_cols_per_op

()

src/gadgets/mul_pairs.rs:27

↓ 3 callersMethodnum_cols_per_op

()

src/gadgets/add_pairs.rs:27

↓ 3 callersMethodnum_cols_per_op

()

src/gadgets/sub_pairs.rs:27

↓ 3 callersMethodnum_cols_per_op

()

src/gadgets/squared_diff.rs:27

↓ 3 callersMethodnum_cols_per_op

()

src/gadgets/max.rs:27

↓ 3 callersFunctionpad

TODO: figure out where to put this

src/layers/shape/pad.rs:18

↓ 3 callersMethodvalid_activations

(self)

python/converter.py:52

↓ 2 callersMethodassign_tensors_map

( &self, mut layouter: impl Layouter<F>, columns: &Vec<Column<Advice>>, tensors: &BTreeMap<i64

src/model.rs:103

↓ 2 callersMethodavg_forward

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, constants: &HashMap

src/layers/averager.rs:25

↓ 2 callersFunctionbroadcast

( x1: &Array<G, IxDyn>, x2: &Array<G, IxDyn>, )

src/utils/helpers.rs:97

↓ 2 callersMethodcommit

( &self, mut layouter: impl Layouter<F>, _gadget_config: Rc<GadgetConfig>, _constants: &HashMa

src/commitments/poseidon_commit.rs:106

↓ 2 callersFunctionconvert_to_bigint

Conversion / printing functions

src/utils/helpers.rs:15

↓ 2 callersMethodget_activation

(&self, layer_params: &Vec<i64>)

src/layers/fully_connected.rs:102

↓ 2 callersMethodget_activation

(&self, layer_params: &Vec<i64>)

src/layers/arithmetic/add.rs:28

↓ 2 callersFunctionget_inputs

(op: tflite.Operator)

python/converter.py:20

↓ 2 callersMethodget_keep_axis

(&self, layer_config: &LayerConfig)

src/layers/mean.rs:20

↓ 2 callersFunctionget_kzg_params

(params_dir: &str, degree: u32)

src/utils/proving_kzg.rs:28

↓ 2 callersMethodget_map

(&self)

src/gadgets/nonlinear/pow.rs:49

↓ 2 callersFunctionload_model_msgpack

(config_path: &str, inp_path: &str)

src/utils/loader.rs:48

↓ 2 callersMethodnum_cols_per_op

()

src/gadgets/bias_div_floor_relu6.rs:44

↓ 2 callersMethodnum_cols_per_op

()

src/gadgets/var_div.rs:28

↓ 2 callersMethodnum_cols_per_op

()

src/gadgets/var_div_big3.rs:26

↓ 2 callersMethodnum_cols_per_op

()

src/gadgets/var_div_big.rs:26

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/square.rs:56

↓ 2 callersMethodnum_cols_per_op

()

src/gadgets/sqrt_big.rs:29

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/nonlinear/rsqrt.rs:60

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/nonlinear/logistic.rs:64

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/nonlinear/pow.rs:63

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/nonlinear/exp.rs:62

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/nonlinear/relu.rs:58

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/nonlinear/sqrt.rs:59

↓ 2 callersMethodnum_cols_per_op

(&self)

src/gadgets/nonlinear/tanh.rs:62

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/bias_div_floor_relu6.rs:133

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/var_div.rs:110

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/var_div_big3.rs:138

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/var_div_big.rs:125

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/bias_div_round_relu6.rs:132

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/square.rs:60

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/sqrt_big.rs:107

↓ 2 callersMethodnum_inputs_per_row

(&self)

src/gadgets/adder.rs:67

↓ 2 callersFunctionshape_dominates

Broadcast

src/utils/helpers.rs:66

↓ 2 callersMethodsplat

( &self, tensors: &Vec<Array<Rc<G>, IxDyn>>, zero: Rc<G>, )

src/layers/conv2d.rs:126

↓ 2 callersMethodtensor_map_to_vec

( &self, tensor_map: &BTreeMap<i64, Array<CellRc<F>, IxDyn>>, )

src/model.rs:142

↓ 2 callersFunctionverify_kzg

( params: &ParamsKZG<Bn256>, vk: &VerifyingKey<G1Affine>, strategy: SingleStrategy<Bn256>, public_vals

src/utils/proving_kzg.rs:54

↓ 1 callersMethod_convert_add

(self, op: tflite.Operator, generated_tensors: set)

python/converter.py:59

↓ 1 callersMethodassign_and_commit

( &self, mut layouter: impl Layouter<F>, constants: &HashMap<i64, CellRc<F>>, config: &ModelCo

src/model.rs:482

↓ 1 callersMethodassign_and_pack

( &self, mut layouter: impl Layouter<F>, gadget_config: Rc<GadgetConfig>, constants: &HashMap<

src/commitments/packer.rs:292

↓ 1 callersMethodassign_and_pack_row

( &self, mut layouter: impl Layouter<F>, gadget_config: Rc<GadgetConfig>, values: Vec<&F>,

src/commitments/packer.rs:213

↓ 1 callersMethodassign_constants

( &self, mut layouter: impl Layouter<F>, gadget_config: Rc<GadgetConfig>, )

src/model.rs:182

↓ 1 callersMethodassign_constants2

TODO: for some horrifying reason, assigning to fixed columns causes everything to blow up Currently get around this by assigning to advice columns Thi

src/model.rs:233

↓ 1 callersMethodassign_tensors_vec

( &self, mut layouter: impl Layouter<F>, columns: &Vec<Column<Advice>>, tensors: &BTreeMap<i64

src/model.rs:166

↓ 1 callersMethodbackward

(self, layer, transcript, config)

backwards/backward.py:252

↓ 1 callersFunctionconvert_pos_int

(x: Value<F>)

src/utils/helpers.rs:23

↓ 1 callersMethodcopy_and_commit

( &self, mut layouter: impl Layouter<F>, constants: &HashMap<i64, CellRc<F>>, config: &ModelCo

src/model.rs:520

↓ 1 callersMethodcopy_and_pack

( &self, mut layouter: impl Layouter<F>, gadget_config: Rc<GadgetConfig>, constants: &HashMap<

src/commitments/packer.rs:339

↓ 1 callersMethodcopy_and_pack_row

( &self, mut layouter: impl Layouter<F>, gadget_config: Rc<GadgetConfig>, cells: Vec<CellRc<F>

src/commitments/packer.rs:142

↓ 1 callersFunctionfinal_shape

(s1: &[usize], s2: &[usize])

src/utils/helpers.rs:89

↓ 1 callersMethodforward

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, constants: &HashMap

src/layers/arithmetic/div_var.rs:44

↓ 1 callersMethodforward

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, constants: &HashMap

src/layers/arithmetic/add.rs:53

↓ 1 callersMethodforward

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, constants: &HashMap

src/layers/arithmetic/sub.rs:41

↓ 1 callersMethodforward

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, constants: &HashMap

src/layers/arithmetic/mul.rs:48

↓ 1 callersMethodgadget_forward

( &self, mut layouter: impl Layouter<F>, vec_inputs: &Vec<Vec<&AssignedCell<F, F>>>, constants

src/layers/arithmetic/add.rs:39

↓ 1 callersMethodget_div_val

( &self, mut layouter: impl Layouter<F>, tensors: &Vec<AssignedTensor<F>>, gadget_config: Rc<G

src/layers/mean.rs:75

↓ 1 callersMethodget_div_val

( &self, mut layouter: impl Layouter<F>, _tensors: &Vec<AssignedTensor<F>>, gadget_config: Rc<

src/layers/div_fixed.rs:21

↓ 1 callersFunctionget_ipa_params

(params_dir: &str, degree: u32)

src/utils/proving_ipa.rs:28

↓ 1 callersMethodget_selector

(&self)

src/gadgets/nonlinear/pow.rs:53

↓ 1 callersFunctionintermediate_shape

Precondition: s1.len() < s2.len()

src/utils/helpers.rs:81

↓ 1 callersFunctionmain

()

python/converter.py:508

↓ 1 callersFunctionmain

()

python/input_converter.py:6

↓ 1 callersFunctionmain

()

python/training_converter.py:18

↓ 1 callersMethodnew_label_tensor

(self)

backwards/backward.py:32

↓ 1 callersMethodnum_inputs_per_row

(&self)

src/gadgets/mul_pairs.rs:71

↓ 1 callersMethodnum_inputs_per_row

(&self)

src/gadgets/add_pairs.rs:71

↓ 1 callersMethodnum_inputs_per_row

(&self)

src/gadgets/sub_pairs.rs:71

↓ 1 callersMethodnum_inputs_per_row

(&self)

src/gadgets/squared_diff.rs:71

↓ 1 callersMethodnum_inputs_per_row

(&self)

src/gadgets/nonlinear/pow.rs:67

↓ 1 callersMethodnum_outputs_per_row

(&self)

src/gadgets/max.rs:96

↓ 1 callersMethodop_row_region

The caller is expected to pad the inputs

src/gadgets/dot_prod.rs:93

↓ 1 callersMethodop_row_region

( &self, region: &mut Region<F>, row_offset: usize, vec_inputs: &Vec<Vec<&AssignedCell<F, F>>>

src/gadgets/max.rs:100

↓ 1 callersFunctionprint_assigned_arr

( prefix: &str, arr: &Vec<&AssignedCell<F, F>>, scale_factor: u64, )

src/utils/helpers.rs:41

↓ 1 callersFunctionprint_pos_int

(prefix: &str, x: Value<F>, scale_factor: u64)

src/utils/helpers.rs:35

next →1–100 of 400, ranked by callers