Summary

Types

dtype/0

input_spec/0

output_spec/0

session/0

tensor/0

load(Path)

run(_, Inputs)

to_list(_)

unload(_)

-type dtype() :: f32 | f64 | i32 | i64 | u8.

-type input_spec() :: {binary(), [integer()], dtype()}.

-type output_spec() :: {binary(), [integer()], dtype()}.

-type session() :: #{ref := reference(), inputs := [input_spec()], outputs := [output_spec()]}.

-type tensor() :: {binary(), [integer()], dtype()}.

-spec load(file:filename()) -> {ok, session()} | {error, term()}.

-spec run(session(), #{binary() => tensor()}) -> {ok, #{binary() => tensor()}} | {error, term()}.

-spec tensor(binary(), [integer()], dtype()) -> tensor().

-spec to_list(tensor()) -> [number()].

-spec unload(session()) -> ok.