Home
last modified time | relevance | path

Searched refs:getOutputTensorCount (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DInterpreter.java559 public int getOutputTensorCount() { in getOutputTensorCount() method in Interpreter
561 return wrapper.getOutputTensorCount(); in getOutputTensorCount()
DNativeInterpreterWrapper.java309 int outputCount = getOutputTensorCount(); in initTensorIndexesMaps()
410 int getOutputTensorCount() { in getOutputTensorCount() method in NativeInterpreterWrapper
/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
DInterpreterFlexTest.java76 assertThat(interpreter.getOutputTensorCount()).isEqualTo(2); in testCommon()
DInterpreterTest.java69 assertThat(interpreter.getOutputTensorCount()).isEqualTo(1); in testInterpreter()
88 assertThat(interpreter.getOutputTensorCount()).isEqualTo(1); in testInterpreterWithOptions()
187 assertThat(interpreter.getOutputTensorCount()).isEqualTo(2); in testRunForMultipleInputsOutputs()