Skip to content

Instantly share code, notes, and snippets.

@miquelbeltran
Created August 31, 2018 11:51
Show Gist options
  • Select an option

  • Save miquelbeltran/d44fabb0e530ab39d4e8bedc4e09beeb to your computer and use it in GitHub Desktop.

Select an option

Save miquelbeltran/d44fabb0e530ab39d4e8bedc4e09beeb to your computer and use it in GitHub Desktop.
// Input is a 1x1 Tensor
val inputDims = intArrayOf(1, 1)
// Output is a 1x1 Tensor
val outputDims = intArrayOf(1, 1)
// Define the Input and Output dimensions and types
val dataOptions = FirebaseModelInputOutputOptions.Builder()
.setInputFormat(0, FirebaseModelDataType.FLOAT32, inputDims)
.setOutputFormat(0, FirebaseModelDataType.FLOAT32, outputDims)
.build()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment