View Source TFLiteElixir.TFLiteTensor (tflite_elixir v0.3.7)

A typed multi-dimensional array used in Tensorflow Lite.

Summary

Types

nif_error()

nif_resource_ok()

tensor_type()

Functions

dims(self)

Get the dimensions (C++) API

quantization_params(self)

Get the quantization params

set_data(self, data)

Set tensor data

shape(self)

Get the tensor shape

to_binary(self, limit \\ 0)

Get binary data

to_nx(self_struct, opts \\ [])

Convert TFLiteElixir.TFLiteTensor to Nx.Tensor

type(self)

Get the data type

Types

nif_error()

@type nif_error() :: {:error, String.t()}

nif_resource_ok()

@type nif_resource_ok() :: {:ok, reference()}

tensor_type()

@type tensor_type() ::
  :no_type
  | {:f, 32}
  | {:s, 32}
  | {:u, 8}
  | {:s, 64}
  | :string
  | :bool
  | {:s, 16}
  | {:c, 64}
  | {:s, 8}
  | {:f, 16}
  | {:f, 64}
  | {:c, 128}
  | {:u, 64}
  | :resource
  | :variant
  | {:u, 32}

Functions

dims(self)

@spec dims(%TFLiteElixir.TFLiteTensor{
  index: term(),
  name: term(),
  quantization_params: term(),
  reference: term(),
  shape: term(),
  shape_signature: term(),
  sparsity_params: term(),
  type: term()
}) :: [integer()]

@spec dims(reference()) :: [integer()] | nif_error()

Get the dimensions (C++) API

quantization_params(self)

@spec quantization_params(
  %TFLiteElixir.TFLiteTensor{
    index: term(),
    name: term(),
    quantization_params: term(),
    reference: term(),
    shape: term(),
    shape_signature: term(),
    sparsity_params: term(),
    type: term()
  }
  | reference()
) ::
  %TFLiteElixir.TFLiteQuantizationParams{
    quantized_dimension: term(),
    scale: term(),
    zero_point: term()
  }
  | nif_error()

Get the quantization params

set_data(self, data)

@spec set_data(
  %TFLiteElixir.TFLiteTensor{
    index: term(),
    name: term(),
    quantization_params: term(),
    reference: term(),
    shape: term(),
    shape_signature: term(),
    sparsity_params: term(),
    type: term()
  }
  | reference(),
  binary()
  | %Nx.Tensor{
      data: term(),
      names: term(),
      shape: term(),
      type: term(),
      vectorized_axes: term()
    }
) :: :ok | nif_error()

Set tensor data

shape(self)

@spec shape(%TFLiteElixir.TFLiteTensor{
  index: term(),
  name: term(),
  quantization_params: term(),
  reference: term(),
  shape: term(),
  shape_signature: term(),
  sparsity_params: term(),
  type: term()
}) :: tuple()

@spec shape(reference()) :: tuple() | nif_error()

Get the tensor shape

to_binary(self, limit \\ 0)

@spec to_binary(
  %TFLiteElixir.TFLiteTensor{
    index: term(),
    name: term(),
    quantization_params: term(),
    reference: term(),
    shape: term(),
    shape_signature: term(),
    sparsity_params: term(),
    type: term()
  }
  | reference(),
  non_neg_integer()
) :: binary() | {:error, String.t()}

Get binary data

to_nx(self_struct, opts \\ [])

@spec to_nx(
  reference()
  | %TFLiteElixir.TFLiteTensor{
      index: term(),
      name: term(),
      quantization_params: term(),
      reference: term(),
      shape: term(),
      shape_signature: term(),
      sparsity_params: term(),
      type: term()
    },
  Keyword.t()
) :: %Nx.Tensor{
  data: term(),
  names: term(),
  shape: term(),
  type: term(),
  vectorized_axes: term()
}

Convert TFLiteElixir.TFLiteTensor to Nx.Tensor

type(self)

@spec type(%TFLiteElixir.TFLiteTensor{
  index: term(),
  name: term(),
  quantization_params: term(),
  reference: term(),
  shape: term(),
  shape_signature: term(),
  sparsity_params: term(),
  type: term()
}) :: tensor_type()

@spec type(reference()) :: tensor_type() | nif_error()

Get the data type