Home
last modified time | relevance | path

Searched defs:inputTensors (Results 1 – 25 of 85) sorted by relevance

1234

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DEndToEndTestImpl.hpp68 InputTensors inputTensors in ConstantUsageTest() local
152 InputTensors inputTensors; in EndToEndLayerTestImpl() local
240 InputTensors inputTensors in ImportNonAlignedInputPointerTest() local
309 InputTensors inputTensors in ExportNonAlignedOutputPointerTest() local
383 InputTensors inputTensors in ImportAlignedPointerTest() local
472 InputTensors inputTensors in ImportOnlyWorkload() local
562 InputTensors inputTensors in ExportOnlyWorkload() local
652 InputTensors inputTensors in ImportAndExportWorkload() local
744 InputTensors inputTensors in ExportOutputWithSeveralOutputSlotConnectionsTest() local
885 InputTensors inputTensors in ForceImportWithAlignedBuffersEndToEndTest() local
[all …]
H A DStridedSliceAsyncEndToEndTest.hpp57 InputTensors inputTensors; in AsyncThreadedEndToEndTestImpl() local
90 InputTensors& inputTensors = inputTensorsVec[i]; in AsyncThreadedEndToEndTestImpl() local
149 InputTensors inputTensors; in AsyncEndToEndTestImpl() local
387 std::vector<std::map<int, std::vector<T>>> inputTensors; in StridedSlicedMultiThreadedEndToEndTest() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroup.hpp217 std::vector<at::Tensor>& inputTensors, in allgather()
268 std::vector<at::Tensor>& inputTensors, in allgather_coalesced()
289 std::vector<at::Tensor>& inputTensors, in allgather_into_tensor_coalesced()
307 std::vector<at::Tensor>& inputTensors, in gather()
327 std::vector<std::vector<at::Tensor>>& inputTensors, in scatter()
351 std::vector<std::vector<at::Tensor>>& inputTensors, in reduce_scatter()
399 std::vector<at::Tensor>& inputTensors, in reduce_scatter_tensor_coalesced()
445 std::vector<at::Tensor>& inputTensors, in alltoall()
H A DProcessGroupMPI.cpp124 const std::optional<std::vector<at::Tensor>>& inputTensors) in AsyncWork()
382 const std::optional<std::vector<at::Tensor>>& inputTensors) { in enqueue()
481 std::vector<at::Tensor>& inputTensors, in allgather()
537 std::vector<at::Tensor>& inputTensors, in gather()
614 std::vector<std::vector<at::Tensor>>& inputTensors, in scatter()
691 std::vector<std::vector<at::Tensor>>& inputTensors, in reduce_scatter()
730 std::vector<at::Tensor> inputTensors = {inputTensor}; in alltoall_base() local
768 std::vector<at::Tensor> inputTensors = {inputTensor}; in alltoall_base() local
781 std::vector<at::Tensor>& inputTensors, in alltoall()
H A DProcessGroupWrapper.cpp433 std::vector<at::Tensor>& inputTensors, in allgather()
447 std::vector<at::Tensor> inputTensors({inputBuffer}); in _allgather_base() local
454 std::vector<at::Tensor>& inputTensors, in allgather_coalesced()
466 std::vector<at::Tensor>& inputTensors, in gather()
474 std::vector<std::vector<at::Tensor>>& inputTensors, in scatter()
482 std::vector<std::vector<at::Tensor>>& inputTensors, in reduce_scatter()
506 std::vector<at::Tensor>& inputTensors, in alltoall()
H A DPyProcessGroup.hpp71 std::vector<at::Tensor>& inputTensors, in allgather()
84 std::vector<at::Tensor>& inputTensors, in allgather_into_tensor_coalesced()
157 std::vector<std::vector<at::Tensor>>& inputTensors, in reduce_scatter()
170 std::vector<at::Tensor>& inputTensors, in reduce_scatter_tensor_coalesced()
H A DProcessGroupUCC.cpp771 std::vector<at::Tensor>& inputTensors, in collective_post()
857 std::vector<at::Tensor>& inputTensors, in allgather()
980 std::vector<at::Tensor> inputTensors = {inputTensor}; in _allgather_base() local
1040 std::vector<at::Tensor>& inputTensors, in alltoall()
1163 std::vector<at::Tensor> inputTensors = {inputTensor}; in alltoall_base() local
1264 std::vector<at::Tensor>& inputTensors, in gather()
1380 std::vector<std::vector<at::Tensor>>& inputTensors, in reduce_scatter()
1454 std::vector<std::vector<at::Tensor>>& inputTensors, in scatter()
/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonFallbackTests.cpp118 InputTensors inputTensors variable
255 InputTensors inputTensors variable
399 InputTensors inputTensors variable
536 InputTensors inputTensors variable
677 InputTensors inputTensors variable
830 InputTensors inputTensors variable
963 InputTensors inputTensors variable
1122 InputTensors inputTensors variable
1268 InputTensors inputTensors variable
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dcpp_c10d_extension.cpp52 std::vector<at::Tensor>& inputTensors, in allgather()
71 std::vector<at::Tensor>& inputTensors, in gather()
78 std::vector<std::vector<at::Tensor>>& inputTensors, in scatter()
85 std::vector<std::vector<at::Tensor>>& inputTensors, in reduce_scatter()
/aosp_15_r20/external/armnn/src/armnn/
H A DRuntime.cpp92 …ctor<ImportedInputId> IRuntime::ImportInputs(NetworkId networkId, const InputTensors& inputTensors, in ImportInputs()
114 const InputTensors& inputTensors, in EnqueueWorkload()
124 const InputTensors& inputTensors, in Execute()
623 …r<ImportedInputId> RuntimeImpl::ImportInputs(NetworkId networkId, const InputTensors& inputTensors, in ImportInputs()
645 const InputTensors& inputTensors, in EnqueueWorkload()
697 const InputTensors& inputTensors, in Execute()
H A DThreadpool.cpp70 const InputTensors& inputTensors, in Schedule()
181 auto inputTensors = std::get<1>(*currentExecInProgress); in ProcessExecPriorities() local
H A DLoadedNetwork.cpp793 WorkloadData(const InputTensors& inputTensors, const OutputTensors& outputTensors) in WorkloadData()
839 Status LoadedNetwork::EnqueueWorkload(const InputTensors& inputTensors, in EnqueueWorkload()
1392 …t armnn::ConstTensor GetInputTensor(const LayerBindingId layerId, const InputTensors& inputTensors) in GetInputTensor()
1418 std::vector<ImportedInputId> LoadedNetwork::ImportInputs(const InputTensors& inputTensors, in ImportInputs()
1727 Status LoadedNetwork::Execute(const InputTensors& inputTensors, in Execute()
/aosp_15_r20/external/armnn/src/backends/cl/test/
H A DClImportTensorHandleTests.cpp177 InputTensors inputTensors variable
386 InputTensors inputTensors variable
528 InputTensors inputTensors variable
676 InputTensors inputTensors variable
798 InputTensors inputTensors variable
961 InputTensors inputTensors variable
1185 InputTensors inputTensors variable
H A DClFallbackTests.cpp126 InputTensors inputTensors variable
251 InputTensors inputTensors variable
408 InputTensors inputTensors variable
545 InputTensors inputTensors variable
/aosp_15_r20/external/armnn/src/backends/reference/test/
H A DRefEndToEndTests.cpp154 armnn::InputTensors inputTensors variable
222 InputTensors inputTensors variable
318 InputTensors inputTensors variable
384 InputTensors inputTensors variable
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/
H A DParserFlatbuffersFixture.hpp293 armnn::InputTensors& inputTensors, in FillInputTensors()
347 armnn::InputTensors inputTensors; in RunTest() local
418 armnn::InputTensors inputTensors; in RunTest() local
467 armnn::InputTensors inputTensors; in RunTest() local
H A DFullyConnected.cpp23 std::string inputTensors = "[ 0, 2 ]"; in FullyConnectedFixture() local
204 std::string inputTensors = "[ 0, 1 ]"; in FullyConnectedNonConstWeightsFixture() local
H A DDepthwiseConvolution2D.cpp22 std::string inputTensors = "[ 0, 2 ]"; in DepthwiseConvolution2dFixture() local
238 std::string inputTensors = "[ 0, 2 ]"; in DepthwiseConvolution2dFixture2() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DArmnnPreparedModel.cpp331 auto inputTensors = std::make_shared<armnn::InputTensors>(); in execute() local
377 armnn::InputTensors& inputTensors, in ExecuteGraph()
534 auto inputTensors = std::make_shared<armnn::InputTensors>(); in executeFenced() local
654 armnn::InputTensors inputTensors; in ExecuteWithDummyInputs() local
/aosp_15_r20/external/android-nn-driver/
H A DArmnnPreparedModel.cpp316 armnn::InputTensors& inputTensors, in ExecuteGraph()
401 armnn::InputTensors inputTensors; in ExecuteWithDummyInputs() local
474 std::shared_ptr<armnn::InputTensors>& inputTensors, in ScheduleGraphForExecution()
H A DArmnnPreparedModel_1_2.cpp526 armnn::InputTensors& inputTensors, in ExecuteGraph()
635 armnn::InputTensors inputTensors; in ExecuteWithDummyInputs() local
704 auto inputTensors = std::make_shared<armnn::InputTensors>(); in Execute() local
767 std::shared_ptr<armnn::InputTensors>& inputTensors, in ScheduleGraphForExecution()
H A DArmnnPreparedModel_1_3.cpp828 armnn::InputTensors& inputTensors, in ExecuteGraph()
933 std::shared_ptr<armnn::InputTensors>& inputTensors, in ScheduleGraphForExecution()
973 armnn::InputTensors inputTensors; in ExecuteWithDummyInputs() local
1044 auto inputTensors = std::make_shared<armnn::InputTensors>(); in Execute() local
/aosp_15_r20/external/armnn/src/profiling/test/
H A DFileOnlyProfilingDecoratorTests.cpp98 InputTensors inputTensors variable
223 InputTensors inputTensors variable
/aosp_15_r20/external/armnn/src/armnn/test/
H A DRuntimeTests.cpp241 InputTensors inputTensors = {{0, inputTensor1}, {1, inputTensor2}}; variable
927 InputTensors inputTensors variable
1542 InputTensors inputTensors{ { 1, inputTensor2 } }; variable
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DFoldPadTests.cpp603 InputTensors inputTensors{{0, ConstTensor(inputTensorInfo, inputData.data())}}; variable
727 InputTensors inputTensors{{0, ConstTensor(inputTensorInfo, inputData.data())}}; variable
852 InputTensors inputTensors{{0, ConstTensor(inputTensorInfo, inputData.data())}}; variable

1234