Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
H A DInterpreterApiTest.java77 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()
90 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()
101 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithNullOptions()
123 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRuntimeFromApplicationOnly()
145 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRuntimeFromSystemOnly()
167 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRuntimePreferSystemOverApplication()
264 assertThat(interpreter.getInputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()
265 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
266 assertThat(interpreter.getInputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
267 assertThat(interpreter.getInputTensor(2).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
[all …]
H A DInterpreterTest.java78 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()
97 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()
108 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithNullOptions()
204 assertThat(interpreter.getInputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()
205 assertThat(interpreter.getInputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
206 assertThat(interpreter.getInputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
207 assertThat(interpreter.getInputTensor(2).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
208 assertThat(interpreter.getInputTensor(3).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
262 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(inputDims); in testResizeInput()
301 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(inputDims); in testUnknownDims()
[all …]
H A 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()
H A 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()
H A DTensorTest.java498 TensorImpl stringTensor = wrapper.getInputTensor(0); in testByteArrayStringTensorInput()
H A DNativeInterpreterWrapperTest.java619 assertThat(wrapper.getInputTensor(0).shape()).isEqualTo(expectedDims); in testGetInputDims()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
H A DNativeInterpreterWrapper.java191 TensorImpl tensor = getInputTensor(input.getKey(), signatureKey); in runSignature()
201 signatureRunnerWrapper.getInputTensor(input.getKey()).setTo(input.getValue()); in runSignature()
233 TensorImpl tensor = getInputTensor(i); in run()
243 getInputTensor(i).setTo(inputs[i]); in run()
371 TensorImpl getInputTensor(int index) { in getInputTensor() method in NativeInterpreterWrapper
390 TensorImpl getInputTensor(String inputName, String signatureKey) { in getInputTensor() method in NativeInterpreterWrapper
397 return signatureRunnerWrapper.getInputTensor(inputName); in getInputTensor()
401 return getInputTensor(inputIndex); in getInputTensor()
H A DInterpreterImpl.java141 public Tensor getInputTensor(int inputIndex) { in getInputTensor() method in InterpreterImpl
143 return wrapper.getInputTensor(inputIndex); in getInputTensor()
H A DNativeSignatureRunnerWrapper.java48 public TensorImpl getInputTensor(String inputName) { in getInputTensor() method in NativeSignatureRunnerWrapper
H A DInterpreterApi.java467 Tensor getInputTensor(int inputIndex); in getInputTensor() method
H A DInterpreter.java304 return wrapper.getInputTensor(inputName, signatureKey); in getInputTensorFromSignature()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/java/tensorflowlite_java/android_common_apex30/turbine/
Dtensorflowlite_java.jarorg/tensorflow/lite/nnapi/NnApiDelegate$PrivateInterface.class <Unknown> package org. ...
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/java/tensorflowlite_java/android_common_apex33/turbine/
Dtensorflowlite_java.jarorg/tensorflow/lite/nnapi/NnApiDelegate$PrivateInterface.class <Unknown> package org. ...
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/java/tensorflowlite_java/android_common_apex30/javac/
Dtensorflowlite_java.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/tensorflow/ org/ ...
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/java/tensorflowlite_java/android_common_apex33/javac/
Dtensorflowlite_java.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/tensorflow/ org/ ...
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/testhelper/java/org/tensorflow/lite/
H A DTestHelper.java46 return interpreter.wrapper.getInputTensor(index).shape(); in getInputDims()
65 return DataTypeUtils.toStringName(interpreter.wrapper.getInputTensor(index).dataType()); in getInputDataType()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/model/
H A DModel.java227 public Tensor getInputTensor(int inputIndex) { in getInputTensor() method in Model
228 return interpreter.getInputTensor(inputIndex); in getInputTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/gpu/
H A DGpuDelegateTest.java94 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(new int[] {1, 224, 224, 3}); in testInterpreterWithGpu_QuantModelRunWithDelegate()
116 assertThat(interpreter.getInputTensor(0).shape()).isEqualTo(new int[] {1, 224, 224, 3}); in testInterpreterWithGpu_QuantModelRunOnCPU()
/aosp_15_r20/external/pytorch/test/cpp/c10d/
H A DProcessGroupNCCLTest.cpp343 at::Tensor getInputTensor() { in getInputTensor() function in AllgatherBaseNCCLTest
398 at::Tensor getInputTensor() { in getInputTensor() function in ReduceScatterBaseNCCLTest
618 auto input_tensor = test.getInputTensor(); in testAllgatherBase()
639 auto input_tensor = test.getInputTensor(); in testReduceScatterBase()
/aosp_15_r20/packages/modules/OnDevicePersonalization/src/com/android/ondevicepersonalization/services/inference/
DIsolatedModelServiceImpl.java155 Tensor tensor = interpreter.getInputTensor(i); in runTfliteInterpreter()
/aosp_15_r20/out/soong/.intermediates/external/tflite-support/tflite_support_java/android_common_apex30/turbine/
Dtflite_support_java.jarorg/tensorflow/lite/support/common/FileUtil.class <Unknown> package org. ...
/aosp_15_r20/out/soong/.intermediates/external/tflite-support/tflite_support_java/android_common_apex33/turbine/
Dtflite_support_java.jarorg/tensorflow/lite/support/common/FileUtil.class <Unknown> package org. ...
/aosp_15_r20/out/soong/.intermediates/external/tflite-support/tflite_support_java/android_common_apex33/javac/
Dtflite_support_java.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/tensorflow/ org/ ...
/aosp_15_r20/out/soong/.intermediates/external/tflite-support/tflite_support_java/android_common_apex30/javac/
Dtflite_support_java.jarMETA-INF/ META-INF/MANIFEST.MF org/ org/tensorflow/ org/ ...
/aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/android/
H A Dquickstart.md230 val inputShape = tflite.getInputTensor(inputIndex).shape()

12