Home
last modified time | relevance | path

Searched refs:getOutputTensor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
DInterpreterTest.java70 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()
89 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()
188 assertThat(interpreter.getOutputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()
189 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
190 assertThat(interpreter.getOutputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
243 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(inputDims); in testResizeInput()
255 assertThat(interpreter.getOutputTensor(0).shape()).isNotEqualTo(inputDims); in testAllocateTensors()
257 assertThat(interpreter.getOutputTensor(0).shape()).isNotEqualTo(inputDims); in testAllocateTensors()
259 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(inputDims); in testAllocateTensors()
263 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(inputDims); in testAllocateTensors()
[all …]
DTensorTest.java69 tensor = wrapper.getOutputTensor(0); in setUp()
161 tensor = wrapper.getOutputTensor(0); in testCopyToIntBuffer()
170 tensor = wrapper.getOutputTensor(0); in testCopyToLongBuffer()
283 tensor = wrapper.getOutputTensor(0); in testSetToInt()
321 tensor = wrapper.getOutputTensor(0); in testSetToLong()
525 tensor = wrapper.getOutputTensor(0); in testQuantizationParameters_quantizedModel()
DInterpreterFlexTest.java77 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
78 assertThat(interpreter.getOutputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
DInterpreterMobileNetTest.java91 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(new int[] {1, 1001}); in runMobileNetFloatTest()
108 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(new int[] {1, 1001}); in runMobileNetQuantizedTest()
/external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/model/
DModel.java236 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor() method in Model
237 return interpreter.getOutputTensor(outputIndex); in getOutputTensor()
246 return interpreter.getOutputTensor(outputIndex).shape(); in getOutputTensorShape()
/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DInterpreter.java588 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor() method in Interpreter
590 return wrapper.getOutputTensor(outputIndex); in getOutputTensor()
622 return wrapper.getOutputTensor(outputName, methodName); in getOutputTensorFromSignature()
DNativeInterpreterWrapper.java216 getOutputTensor(output.getKey()).copyTo(output.getValue()); in run()
419 Tensor getOutputTensor(int index) { in getOutputTensor() method in NativeInterpreterWrapper
437 Tensor getOutputTensor(String outputName, String methodName) { in getOutputTensor() method in NativeInterpreterWrapper
448 return getOutputTensor(tensorToOutputsIndexes.get(tensorIndex)); in getOutputTensor()
/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/gpu/
DGpuDelegateTest.java88 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(new int[] {1, 1001}); in testInterpreterWithGpu_QuantModelRunWithDelegate()
110 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(new int[] {1, 1001}); in testInterpreterWithGpu_QuantModelRunOnCPU()
/external/tensorflow/tensorflow/lite/java/src/testhelper/java/org/tensorflow/lite/
DTestHelper.java84 return interpreter.wrapper.getOutputTensor(index).dataType().toStringName(); in getOutputDataType()