Packages

tflite_elixir

0.1.4
0.3.10 0.3.9 0.3.8 0.3.7 0.3.6 0.3.5 0.3.4 0.3.3 0.3.0 0.1.7 0.1.6 0.1.5 0.1.4 0.1.3 0.1.2

TensorFlow Lite Elixir binding with optional TPU support.

HexDocs

Current section

Files

Jump to
Readme
15 Versions
3 Dependencies
Dependants
Files
Activity
tflite_elixir lib tflite_elixir.ex
Raw

lib/tflite_elixir.ex 

defmodule TFLiteElixir do

  alias TFLiteElixir.TFLiteTensor



  @doc """

  Prints a dump of what tensors and what nodes are in the interpreter.



  Note that this function directly prints to stdout

  """

  @spec print_interpreter_state(reference()) :: nil

  def print_interpreter_state(interpreter) do

    TFLiteElixir.Nif.tflite_printInterpreterState(interpreter)

    nil

  end



  def reset_variable_tensor(%TFLiteTensor{}=tflite_tensor) do

    TFLiteElixir.Nif.tflite_resetVariableTensor(tflite_tensor)

  end

end