Home
last modified time | relevance | path

Searched defs:inputTensor (Results 1 – 25 of 65) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Tests/
H A DInterpreterTests.swift74 let inputTensor = try interpreter.input(at: AddModel.validIndex) in testInputTensorAtIndex() variable
81 let inputTensor = try interpreter.input(at: AddQuantizedModel.inputOutputIndex) in testInputTensorAtIndex_QuantizedModel() variable
162 let inputTensor = try interpreter.copy(AddModel.inputData, toInputAt: AddModel.validIndex) in testCopyDataToInputTensorAtIndex() variable
286 static let inputTensor = Tensor( variable
316 static let inputTensor = Tensor( variable
H A DSignatureRunnerTest.swift35 let inputTensor = try addRunner.input(named: "x") in testResizeInputTensor() variable
124 static let inputTensor = Tensor( variable
H A DTensorTests.swift27 let inputTensor = Tensor( in testInit() variable
/aosp_15_r20/external/armnn/delegate/common/src/test/
H A DDelegateTestInterpreter.hpp68 …TfLiteTensor* inputTensor = delegateTestInterpreter::GetInputTensorFromInterpreter(m_TfLiteInterpr… in FillInputTensor() local
77 …TfLiteTensor* inputTensor = delegateTestInterpreter::GetInputTensorFromInterpreter(m_TfLiteInterpr… in FillInputTensor() local
H A DDelegateTestInterpreterUtils.hpp21 TfLiteTensor* inputTensor = TfLiteInterpreterGetInputTensor(interpreter, index); in GetInputTensorFromInterpreter() local
/aosp_15_r20/external/android-nn-driver/
H A DArmnnPreparedModel.cpp242 …const armnn::Tensor inputTensor = GetTensorForRequestArgument(inputArg, inputTensorInfo, *pMemPool… in execute() local
410 const armnn::ConstTensor inputTensor(inputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
H A DArmnnPreparedModel_1_2.cpp329 …const armnn::Tensor inputTensor = GetTensorForRequestArgument(inputArg, inputTensorInfo, memPools); in PrepareMemoryForInputs() local
644 const armnn::ConstTensor inputTensor(inputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
H A DArmnnPreparedModel_1_3.cpp533 …const armnn::Tensor inputTensor = GetTensorForRequestArgument(inputArg, inputTensorInfo, memPools); in PrepareMemoryForInputs() local
982 const armnn::ConstTensor inputTensor(inputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
/aosp_15_r20/external/armnn/src/armnn/
H A DLoadedNetwork.cpp800 auto inputTensor = inputTensorPair.second; in WorkloadData() local
1304 void LoadedNetwork::EnqueueInput(const ConstTensor& inputTensor, ITensorHandle* inputTensorHandle) in EnqueueInput()
1449 { in ImportInputs()
1459 const auto& inputTensor = *it; in ImportInputs() local
1490 for (auto inputTensor : inputTensors) in ImportInputs() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DArmnnPreparedModel.cpp174 …const armnn::Tensor inputTensor = GetTensorForRequestArgument(inputArg, inputTensorInfo, memPools); in PrepareMemoryForInputs() local
662 const armnn::ConstTensor inputTensor(inputTensorInfo, storage.back().data()); in ExecuteWithDummyInputs() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp3599 at::Tensor& inputTensor, in checkForNCCLErrorsInternal()
3697 at::Tensor& inputTensor, in checkForNCCLErrorsInternal()
3735 auto inputTensor = inputTensors.back(); in checkForNCCLErrorsInternal() local
3966 at::Tensor& inputTensor, in checkForNCCLErrorsInternal()
4150 at::Tensor& inputTensor, in checkForNCCLErrorsInternal()
4438 auto inputTensor = inputTensors.back(); in checkForNCCLErrorsInternal() local
H A DProcessGroupGloo.cpp1945 at::Tensor& inputTensor, in _reduce_scatter_base()
2616 at::Tensor& inputTensor, in AsyncAlltoallWork()
2636 at::Tensor inputTensor; member in c10d::__anon2d5ad4bf1911::AsyncAlltoallWork
2641 void alltoall(at::Tensor& outputTensor, at::Tensor& inputTensor) { in alltoall()
2680 at::Tensor& inputTensor, in AsyncAlltoallCUDAWork()
2740 at::Tensor& inputTensor, in alltoall_base()
/aosp_15_r20/external/executorch/examples/demo-apps/android/ExecuTorchDemo/app/src/main/java/com/example/executorchdemo/
H A DClassificationActivity.java66 final Tensor inputTensor = in run() local
H A DMainActivity.java183 final Tensor inputTensor = in run() local
/aosp_15_r20/external/armnn/src/armnnUtils/
H A DTensorIOUtils.hpp46 armnn::ConstTensor inputTensor(inputTensorInfo, value.data()); in MakeInputTensors() local
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DSlice.hpp34 const TfLiteTensor* inputTensor = &tfLiteTensors[tfLiteNode->inputs->data[i]]; in VisitSliceOperator() local
H A DStridedSlice.hpp34 const TfLiteTensor* inputTensor = &tfLiteTensors[tfLiteNode->inputs->data[i]]; in VisitStridedSliceOperator() local
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp105 …const arm_compute::ICLTensor& inputTensor = static_cast<IClTensorHandle*>(m_Data.m_Inputs[… in ClQuantizedLstmWorkload() local
/aosp_15_r20/external/armnn/delegate/test/
H A DPreluTestHelper.hpp50 auto inputTensor = CreateTensor(flatBufferBuilder, in CreatePreluTfLiteModel() local
H A DRedefineTestHelper.hpp47 auto inputTensor = CreateTensor(flatBufferBuilder, in CreateRedefineTfLiteModel() local
H A DArgMinMaxTestHelper.hpp45 auto inputTensor = CreateTensor(flatBufferBuilder, in CreateArgMinMaxTfLiteModel() local
H A DPadTestHelper.hpp47 auto inputTensor = CreateTensor(flatBufferBuilder, in CreatePadTfLiteModel() local
H A DNormalizationTestHelper.hpp45 auto inputTensor = CreateTensor(flatBufferBuilder, in CreateNormalizationTfLiteModel() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DLogSoftmaxTestImpl.cpp42 auto inputTensor = armnnUtils::QuantizedVector<T>(inputValues, qScale, qOffset); in LogSoftmaxTestImpl() local
H A DInstanceNormalizationTestImpl.cpp39 std::vector<T> inputTensor = armnnUtils::QuantizedVector<T>(inputValues, qScale, qOffset); in InstanceNormTestImpl() local

123