Home
last modified time | relevance | path

Searched refs:getInputTensor (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
DInterpreterFlexTest.java72 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
73 assertThat(interpreter.getInputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
74 assertThat(interpreter.getInputTensor(2).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
75 assertThat(interpreter.getInputTensor(3).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
DInterpreterTest.java68 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()
87 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()
182 assertThat(interpreter.getInputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()
183 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
184 assertThat(interpreter.getInputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
185 assertThat(interpreter.getInputTensor(2).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
186 assertThat(interpreter.getInputTensor(3).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
239 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(inputDims); in testResizeInput()
278 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(inputDims); in testUnknownDims()
279 assertThat(interpreter.getInputTensor(0).shapeSignature()).isEqualTo(inputDimsSignature); in testUnknownDims()
[all …]
DInterpreterMobileNetTest.java90 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(new int[] {1, 224, 224, 3}); in runMobileNetFloatTest()
107 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(new int[] {1, 224, 224, 3}); in runMobileNetQuantizedTest()
DTensorTest.java539 Tensor stringTensor = wrapper.getInputTensor(0); in testByteArrayStringTensorInput()
DNativeInterpreterWrapperTest.java607 assertThat(wrapper.getInputTensor(0).shape()).isEqualTo(expectedDims); in testGetInputDims()
/external/tensorflow/tensorflow/lite/java/src/testhelper/java/org/tensorflow/lite/
DTestHelper.java46 return interpreter.wrapper.getInputTensor(index).shape(); in getInputDims()
65 return interpreter.wrapper.getInputTensor(index).dataType().toStringName(); in getInputDataType()
/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DInterpreter.java498 public Tensor getInputTensor(int inputIndex) { in getInputTensor() method in Interpreter
500 return wrapper.getInputTensor(inputIndex); in getInputTensor()
525 return wrapper.getInputTensor(inputName, methodName); in getInputTensorFromSignature()
DNativeInterpreterWrapper.java186 Tensor tensor = getInputTensor(i); in run()
200 getInputTensor(i).setTo(inputs[i]); in run()
356 Tensor getInputTensor(int index) { in getInputTensor() method in NativeInterpreterWrapper
374 Tensor getInputTensor(String inputName, String methodName) { in getInputTensor() method in NativeInterpreterWrapper
385 return getInputTensor(tensorToInputsIndexes.get(tensorIndex)); in getInputTensor()
/external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/model/
DModel.java227 public Tensor getInputTensor(int inputIndex) { in getInputTensor() method in Model
228 return interpreter.getInputTensor(inputIndex); in getInputTensor()
/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/gpu/
DGpuDelegateTest.java87 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(new int[] {1, 224, 224, 3}); in testInterpreterWithGpu_QuantModelRunWithDelegate()
109 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(new int[] {1, 224, 224, 3}); in testInterpreterWithGpu_QuantModelRunOnCPU()