Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
H A DInterpreterApiTest.java79 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()
92 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()
103 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithNullOptions()
125 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRuntimeFromApplicationOnly()
147 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRuntimeFromSystemOnly()
169 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRuntimePreferSystemOverApplication()
270 assertThat(interpreter.getOutputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()
271 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
272 assertThat(interpreter.getOutputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
326 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(inputDims); in testResizeInput()
[all …]
H A DInterpreterTest.java80 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreter()
99 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithOptions()
110 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testInterpreterWithNullOptions()
210 assertThat(interpreter.getOutputTensor(0).index()).isGreaterThan(-1); in testRunForMultipleInputsOutputs()
211 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
212 assertThat(interpreter.getOutputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testRunForMultipleInputsOutputs()
266 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(inputDims); in testResizeInput()
278 assertThat(interpreter.getOutputTensor(0).shape()).isNotEqualTo(inputDims); in testAllocateTensors()
280 assertThat(interpreter.getOutputTensor(0).shape()).isNotEqualTo(inputDims); in testAllocateTensors()
282 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(inputDims); in testAllocateTensors()
[all …]
H A DTensorTest.java70 tensor = wrapper.getOutputTensor(0); in setUp()
163 tensor = wrapper.getOutputTensor(0); in testCopyToIntBuffer()
172 tensor = wrapper.getOutputTensor(0); in testCopyToLongBuffer()
269 tensor = wrapper.getOutputTensor(0); in testSetToInt()
307 tensor = wrapper.getOutputTensor(0); in testSetToLong()
483 tensor = wrapper.getOutputTensor(0); in testQuantizationParameters_quantizedModel()
H A DInterpreterFlexTest.java77 assertThat(interpreter.getOutputTensor(0).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
78 assertThat(interpreter.getOutputTensor(1).dataType()).isEqualTo(DataType.FLOAT32); in testCommon()
H A 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()
/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/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
H A DNativeInterpreterWrapper.java211 signatureRunnerWrapper.getOutputTensor(output.getKey()).copyTo(output.getValue()); in runSignature()
261 getOutputTensor(output.getKey()).copyTo(output.getValue()); in run()
429 TensorImpl getOutputTensor(int index) { in getOutputTensor() method in NativeInterpreterWrapper
448 TensorImpl getOutputTensor(String outputName, String signatureKey) { in getOutputTensor() method in NativeInterpreterWrapper
455 return signatureRunnerWrapper.getOutputTensor(outputName); in getOutputTensor()
459 return getOutputTensor(outputIndex); in getOutputTensor()
H A DInterpreterImpl.java160 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor() method in InterpreterImpl
162 return wrapper.getOutputTensor(outputIndex); in getOutputTensor()
H A DNativeSignatureRunnerWrapper.java53 public TensorImpl getOutputTensor(String outputName) { in getOutputTensor() method in NativeSignatureRunnerWrapper
H A DInterpreterApi.java493 Tensor getOutputTensor(int outputIndex); in getOutputTensor() method
H A DInterpreter.java366 return wrapper.getOutputTensor(outputName, signatureKey); in getOutputTensorFromSignature()
/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/pytorch/aten/src/ATen/native/quantized/cudnn/
H A DBinaryOps.cpp180 .setxDesc(rhs_mult_op.getOutputTensor()) in add()
193 .setxDesc(add_op.getOutputTensor()) in add()
202 .setxDesc(kReluFused ? relu_op.value().getOutputTensor() : add_op.getOutputTensor()) in add()
H A DLinear.cpp211 .setxDesc(linear_op.getOutputTensor()) in apply_impl_helper()
226 …_ptr = bias_.has_value() ? sum_linear_bias_op.value().getOutputTensor() : linear_op.getOutputTenso… in apply_impl_helper()
241 .setxDesc(kReluFused ? relu_op.value().getOutputTensor() : tensor2requant_ptr) in apply_impl_helper()
H A DConv.cpp180 .setxDesc(conv_op.getOutputTensor()) in apply_impl_helper()
192 …ant_ptr = bias_.has_value() ? sum_conv_bias_op.value().getOutputTensor() : conv_op.getOutputTensor in apply_impl_helper()
207 .setxDesc(kReluFused ? relu_op.value().getOutputTensor() : tensor2requant_ptr) in apply_impl_helper()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/model/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/gpu/
H A DGpuDelegateTest.java95 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(new int[] {1, 1001}); in testInterpreterWithGpu_QuantModelRunWithDelegate()
117 assertThat(interpreter.getOutputTensor(0).shape()).isEqualTo(new int[] {1, 1001}); in testInterpreterWithGpu_QuantModelRunOnCPU()
/aosp_15_r20/external/pytorch/test/cpp/c10d/
H A DProcessGroupNCCLTest.cpp338 at::Tensor getOutputTensor() { in getOutputTensor() function in AllgatherBaseNCCLTest
393 at::Tensor getOutputTensor() { in getOutputTensor() function in ReduceScatterBaseNCCLTest
617 auto output_tensor = test.getOutputTensor(); in testAllgatherBase()
638 auto output_tensor = test.getOutputTensor(); in testReduceScatterBase()
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/testhelper/java/org/tensorflow/lite/
H A DTestHelper.java84 return DataTypeUtils.toStringName(interpreter.wrapper.getOutputTensor(index).dataType()); in getOutputDataType()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConv_v8.cpp513 .setxDesc(conv_op.getOutputTensor()) in build_opgraph_fused()
531 .setxDesc(add_op.getOutputTensor()) in build_opgraph_fused()
547 .setxDesc(add_bias_op.getOutputTensor()) in build_opgraph_fused()
/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/ ...

12