Searched refs:getInputTensor (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/ |
D | InterpreterFlexTest.java | 72 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()
|
D | InterpreterTest.java | 68 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 …]
|
D | InterpreterMobileNetTest.java | 90 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()
|
D | TensorTest.java | 539 Tensor stringTensor = wrapper.getInputTensor(0); in testByteArrayStringTensorInput()
|
D | NativeInterpreterWrapperTest.java | 607 assertThat(wrapper.getInputTensor(0).shape()).isEqualTo(expectedDims); in testGetInputDims()
|
/external/tensorflow/tensorflow/lite/java/src/testhelper/java/org/tensorflow/lite/ |
D | TestHelper.java | 46 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/ |
D | Interpreter.java | 498 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()
|
D | NativeInterpreterWrapper.java | 186 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/ |
D | Model.java | 227 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/ |
D | GpuDelegateTest.java | 87 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()
|