Searched refs:getOutputTensor (Results 1 – 8 of 8) sorted by relevance
47 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testFlexModel()48 assertThat(interpreter.getOutputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testFlexModel()
57 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()69 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()179 assertThat(interpreter.getOutputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()180 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()181 assertThat(interpreter.getOutputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()225 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(inputDims); in testResizeInput()
50 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(new int[] {1, 1001}); in testMobilenetRun()
52 tensor = wrapper.getOutputTensor(0); in setUp()
337 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor() method in Interpreter339 return wrapper.getOutputTensor(outputIndex); in getOutputTensor()
161 getOutputTensor(output.getKey()).copyTo(output.getValue()); in run()299 Tensor getOutputTensor(int index) { in getOutputTensor() method in NativeInterpreterWrapper
84 return interpreter.wrapper.getOutputTensor(index).dataType().toStringName(); in getOutputDataType()
245 Tensor outputTensor = interpreter.getOutputTensor(0);