INNER CODE UNIT · Rust

build

SitinCloud/Owlyshield · owlyshield_predict/moonfire-tflite/src/lib.rs:138

    pub fn build(
        mut self,
        model: &Model,
        seq_len: usize,
        vector_len: usize,
    ) -> Result<Interpreter<'a>, ()> {
        let interpreter =
            unsafe { TfLiteInterpreterCreate(model.0.as_ptr(), self.options.as_ptr()) };
        let interpreter = Interpreter {
            interpreter: ptr::NonNull::new(interpreter).ok_or(())?,
            _owned_delegates: std::mem::take(&mut self.owned_delegates),
            _delegate_refs: PhantomData,
        };
        let input_data = vec![1, seq_len as u32, vector_len as u32];
        unsafe {
            TfLiteInterpreterResizeInputTensor(
                interpreter.interpreter.as_ptr(),
                0,

View source record →