code/bugfind/bench/kernels/tract_len/pre/src/lib.rs

align / code /bugfind /bench /kernels /tract_len /pre /src /lib.rs

Add files using upload-large-folder tool

63f41a6 verified 16 days ago

272 Bytes

pub fn tensor_len_ok(dims: &[usize], bits_per_item: usize, data_size_bytes: usize) -> bool {
    let mut len = 1usize;
    let mut i = 0usize;
    while i < dims.len() {
        len = len * dims[i];
        i += 1;
    }
    len * (bits_per_item / 8) == data_size_bytes
}

	pub fn tensor_len_ok(dims: &[usize], bits_per_item: usize, data_size_bytes: usize) -> bool {
	let mut len = 1usize;
	let mut i = 0usize;
	while i < dims.len() {
	len = len * dims[i];
	i += 1;
	}
	len * (bits_per_item / 8) == data_size_bytes
	}