Searched refs:getInputTensor (Results 1 – 8 of 8) sorted by relevance
42 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testFlexModel()43 assertThat(interpreter.getInputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testFlexModel()44 assertThat(interpreter.getInputTensor(2).dataType()).isEqualTo(DataType.FLOAT32); in testFlexModel()45 assertThat(interpreter.getInputTensor(3).dataType()).isEqualTo(DataType.FLOAT32); in testFlexModel()
55 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()67 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()173 assertThat(interpreter.getInputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()174 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()175 assertThat(interpreter.getInputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()176 assertThat(interpreter.getInputTensor(2).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()177 assertThat(interpreter.getInputTensor(3).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()221 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(inputDims); in testResizeInput()
49 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(new int[] {1, 224, 224, 3}); in testMobilenetRun()
538 assertThat(wrapper.getInputTensor(0).shape()).isEqualTo(expectedDims); in testGetInputDims()
46 return interpreter.wrapper.getInputTensor(index).shape(); in getInputDims()65 return interpreter.wrapper.getInputTensor(index).dataType().toStringName(); in getInputDataType()
310 public Tensor getInputTensor(int inputIndex) { in getInputTensor() method in Interpreter312 return wrapper.getInputTensor(inputIndex); in getInputTensor()
131 Tensor tensor = getInputTensor(i); in run()145 getInputTensor(i).setTo(inputs[i]); in run()276 Tensor getInputTensor(int index) { in getInputTensor() method in NativeInterpreterWrapper
212 Tensor inputTensor = interpreter.getInputTensor(0);