Home
last modified time | relevance | path

Searched defs:outputTensor (Results 1 – 25 of 57) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Tests/
H A DInterpreterTests.swift105 let outputTensor = try interpreter.output(at: AddModel.validIndex) in testOutputTensorAtIndex() variable
115 let outputTensor = try interpreter.output(at: AddQuantizedModel.inputOutputIndex) in testOutputTensorAtIndex_QuantizedModel() variable
292 static let outputTensor = Tensor( variable
323 static let outputTensor = Tensor( variable
H A DSignatureRunnerTest.swift81 let outputTensor = try addRunner.output(named: "output_0") in testInvokeWithInputs() variable
130 static let outputTensor = Tensor( variable
/aosp_15_r20/external/armnn/delegate/common/src/test/
H A DDelegateTestInterpreter.hpp109 const TfLiteTensor* outputTensor = in GetOutputResult() local
130 const TfLiteTensor* outputTensor = in GetOutputResult() local
151 const TfLiteTensor* outputTensor = in GetOutputShape() local
H A DDelegateTestInterpreterUtils.hpp31 const TfLiteTensor* outputTensor = TfLiteInterpreterGetOutputTensor(interpreter, index); in GetOutputTensorFromInterpreter() local
/aosp_15_r20/external/android-nn-driver/
H A DArmnnPreparedModel_1_2.cpp365 …const armnn::Tensor outputTensor = GetTensorForRequestArgument(outputArg, outputTensorInfo, memPoo… in PrepareMemoryForOutputs() local
542 const armnn::Tensor outputTensor = outputTensorPair.second; in ExecuteGraph() local
654 const armnn::Tensor outputTensor(outputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
780 const armnn::Tensor outputTensor = outputTensorPair.second; in ScheduleGraphForExecution() local
H A DArmnnPreparedModel_1_3.cpp569 …const armnn::Tensor outputTensor = GetTensorForRequestArgument(outputArg, outputTensorInfo, memPoo… in PrepareMemoryForOutputs() local
842 const armnn::Tensor outputTensor = outputTensorPair.second; in ExecuteGraph() local
947 const armnn::Tensor outputTensor = outputTensorPair.second; in ScheduleGraphForExecution() local
992 const armnn::Tensor outputTensor(outputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
H A DArmnnPreparedModel.cpp267 …const armnn::Tensor outputTensor = GetTensorForRequestArgument(outputArg, outputTensorInfo, *pMemP… in execute() local
420 const armnn::Tensor outputTensor(outputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DArmnnPreparedModel.cpp199 … armnn::Tensor outputTensor = GetTensorForRequestArgument(outputArg, outputTensorInfo, memPools); in PrepareMemoryForOutputs() local
357 const armnn::Tensor outputTensor = outputTensorPair.second; in execute() local
672 const armnn::Tensor outputTensor(outputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
/aosp_15_r20/external/armnn/delegate/test/
H A DRedefineTestHelper.hpp63 auto outputTensor = CreateTensor(flatBufferBuilder, in CreateRedefineTfLiteModel() local
91 auto outputTensor = CreateTensor(flatBufferBuilder, in CreateRedefineTfLiteModel() local
H A DPreluTestHelper.hpp66 auto outputTensor = CreateTensor(flatBufferBuilder, in CreatePreluTfLiteModel() local
H A DArgMinMaxTestHelper.hpp60 auto outputTensor = CreateTensor(flatBufferBuilder, in CreateArgMinMaxTfLiteModel() local
H A DPadTestHelper.hpp62 auto outputTensor = CreateTensor(flatBufferBuilder, in CreatePadTfLiteModel() local
H A DNormalizationTestHelper.hpp53 auto outputTensor = CreateTensor(flatBufferBuilder, in CreateNormalizationTfLiteModel() local
/aosp_15_r20/external/armnn/src/armnn/
H A DLoadedNetwork.cpp811 auto outputTensor = outputTensorPair.second; in WorkloadData() local
1350 void LoadedNetwork::ImportOutputTensor(const Tensor& outputTensor, ITensorHandle* outputTensorHandl… in ImportOutputTensor()
1376 void CopyToOutputTensor(const Tensor& outputTensor, ITensorHandle* outputTensorHandle) in CopyToOutputTensor()
1590 { in ImportOutputs()
1600 const auto outputTensor = *it; in ImportOutputs() local
1623 for (const auto& outputTensor : outputTensors) in ImportOutputs() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp3347 at::Tensor outputTensor = in checkForNCCLErrorsInternal() local
3598 at::Tensor& outputTensor, in checkForNCCLErrorsInternal()
3696 at::Tensor& outputTensor, in checkForNCCLErrorsInternal()
3863 auto outputTensor = outputTensors.back(); in checkForNCCLErrorsInternal() local
3965 at::Tensor& outputTensor, in checkForNCCLErrorsInternal()
4149 at::Tensor& outputTensor, in checkForNCCLErrorsInternal()
4530 auto outputTensor = outputTensors.back(); in checkForNCCLErrorsInternal() local
H A DProcessGroupGloo.cpp1944 at::Tensor& outputTensor, in _reduce_scatter_base()
2615 at::Tensor& outputTensor, in AsyncAlltoallWork()
2635 at::Tensor outputTensor; member in c10d::__anon2d5ad4bf1911::AsyncAlltoallWork
2641 void alltoall(at::Tensor& outputTensor, at::Tensor& inputTensor) { in alltoall()
2679 at::Tensor& outputTensor, in AsyncAlltoallCUDAWork()
2739 at::Tensor& outputTensor, in alltoall_base()
H A DProcessGroupUCC.cpp958 at::Tensor& outputTensor, in _allgather_base()
1103 at::Tensor& outputTensor, in alltoall_base()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DRuntimeTests.cpp111 Tensor outputTensor({{4}, armnn::DataType::Signed32}, output.data()); variable
1527 Tensor outputTensor({ { 4 }, armnn::DataType::Signed32 }, output.data()); variable
/aosp_15_r20/external/executorch/examples/demo-apps/android/ExecuTorchDemo/app/src/main/java/com/example/executorchdemo/
H A DClassificationActivity.java73 final Tensor outputTensor = module.forward(EValue.from(inputTensor))[0].toTensor(); in run() local
H A DMainActivity.java191 Tensor outputTensor = mModule.forward(EValue.from(inputTensor))[0].toTensor(); in run() local
/aosp_15_r20/external/armnn/src/armnnUtils/
H A DTensorIOUtils.hpp82 armnn::Tensor outputTensor(outputBinding.second, value.data()); in MakeOutputTensors() local
H A DModelAccuracyChecker.hpp73 void AddImageResult(const std::string& imageName, std::vector<TContainer> outputTensor) in AddImageResult()
/aosp_15_r20/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
H A DEagerOperationTest.java57 public void outputTensor() { in outputTensor() method in EagerOperationTest
/aosp_15_r20/external/pytorch/android/test_app/app/src/main/java/org/pytorch/testapp/
H A DMainActivity.java142 final Tensor outputTensor = mModule.forward(IValue.from(mInputTensor)).toTensor(); in doModuleForward() local
H A DCameraActivity.java191 final Tensor outputTensor = mModule.forward(IValue.from(mInputTensor)).toTensor(); in analyzeImage() local

123