| /aosp_15_r20/external/pytorch/test/cpp/c10d/ |
| H A D | ProcessGroupMPITest.cpp | 19 std::vector<std::vector<at::Tensor>> outputTensors; in waitWork() local
36 std::vector<std::vector<at::Tensor>> outputTensors; in waitFuture() local
71 auto outputTensors = waitFuture(pg, works); in testAllreduce() local
106 auto outputTensors = waitFuture(pg, works); in testBroadcast() local
132 auto outputTensors = waitFuture(pg, works); in testReduce() local
174 auto outputTensors = waitFuture(pg, works); in testAllgather() local
216 auto outputTensors = waitFuture(pg, works); in testGather() local
266 auto outputTensors = waitFuture(pg, works); in testScatter() local
314 auto outputTensors = waitWork(pg, works); in testSendRecv() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | EndToEndTestImpl.hpp | 72 OutputTensors outputTensors in ConstantUsageTest() local
159 OutputTensors outputTensors; in EndToEndLayerTestImpl() local
244 OutputTensors outputTensors in ImportNonAlignedInputPointerTest() local
313 OutputTensors outputTensors in ExportNonAlignedOutputPointerTest() local
387 OutputTensors outputTensors in ImportAlignedPointerTest() local
476 OutputTensors outputTensors in ImportOnlyWorkload() local
566 OutputTensors outputTensors in ExportOnlyWorkload() local
656 OutputTensors outputTensors in ImportAndExportWorkload() local
748 OutputTensors outputTensors in ExportOutputWithSeveralOutputSlotConnectionsTest() local
889 OutputTensors outputTensors in ForceImportWithAlignedBuffersEndToEndTest() local
[all …]
|
| H A D | StridedSliceAsyncEndToEndTest.hpp | 58 OutputTensors outputTensors; in AsyncThreadedEndToEndTestImpl() local
91 OutputTensors& outputTensors = outputTensorsVec[i]; in AsyncThreadedEndToEndTestImpl() local
166 OutputTensors outputTensors; in AsyncEndToEndTestImpl() local
388 std::vector<std::map<int, std::vector<T>>> outputTensors; in StridedSlicedMultiThreadedEndToEndTest() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroup.hpp | 216 std::vector<std::vector<at::Tensor>>& outputTensors, in allgather()
288 std::vector<at::Tensor>& outputTensors, in allgather_into_tensor_coalesced()
306 std::vector<std::vector<at::Tensor>>& outputTensors, in gather()
326 std::vector<at::Tensor>& outputTensors, in scatter()
350 std::vector<at::Tensor>& outputTensors, in reduce_scatter()
398 std::vector<at::Tensor>& outputTensors, in reduce_scatter_tensor_coalesced()
444 std::vector<at::Tensor>& outputTensors, in alltoall()
|
| H A D | ProcessGroupMPI.cpp | 122 std::vector<at::Tensor> outputTensors, in AsyncWork()
480 std::vector<std::vector<at::Tensor>>& outputTensors, in allgather()
536 std::vector<std::vector<at::Tensor>>& outputTensors, in gather()
613 std::vector<at::Tensor>& outputTensors, in scatter()
690 std::vector<at::Tensor>& outputTensors, in reduce_scatter()
731 std::vector<at::Tensor> outputTensors = {outputTensor}; in alltoall_base() local
769 std::vector<at::Tensor> outputTensors = {outputTensor}; in alltoall_base() local
780 std::vector<at::Tensor>& outputTensors, in alltoall()
|
| H A D | PyProcessGroup.hpp | 70 std::vector<std::vector<at::Tensor>>& outputTensors, in allgather()
83 std::vector<at::Tensor>& outputTensors, in allgather_into_tensor_coalesced()
156 std::vector<at::Tensor>& outputTensors, in reduce_scatter()
169 std::vector<at::Tensor>& outputTensors, in reduce_scatter_tensor_coalesced()
|
| H A D | ProcessGroupUCC.cpp | 772 std::vector<at::Tensor>& outputTensors, in collective_post()
856 std::vector<std::vector<at::Tensor>>& outputTensors, in allgather()
981 std::vector<at::Tensor> outputTensors = {outputTensor}; in _allgather_base() local
1039 std::vector<at::Tensor>& outputTensors, in alltoall()
1164 std::vector<at::Tensor> outputTensors = {outputTensor}; in alltoall_base() local
1263 std::vector<std::vector<at::Tensor>>& outputTensors, in gather()
1379 std::vector<at::Tensor>& outputTensors, in reduce_scatter()
1453 std::vector<at::Tensor>& outputTensors, in scatter()
|
| H A D | ProcessGroupWrapper.cpp | 432 std::vector<std::vector<at::Tensor>>& outputTensors, in allgather()
465 std::vector<std::vector<at::Tensor>>& outputTensors, in gather()
473 std::vector<at::Tensor>& outputTensors, in scatter()
481 std::vector<at::Tensor>& outputTensors, in reduce_scatter()
505 std::vector<at::Tensor>& outputTensors, in alltoall()
|
| /aosp_15_r20/external/armnn/src/backends/neon/test/ |
| H A D | NeonFallbackTests.cpp | 124 OutputTensors outputTensors variable
260 OutputTensors outputTensors variable
405 OutputTensors outputTensors variable
541 OutputTensors outputTensors variable
683 OutputTensors outputTensors variable
836 OutputTensors outputTensors variable
969 OutputTensors outputTensors variable
1128 OutputTensors outputTensors variable
1274 OutputTensors outputTensors variable
|
| H A D | NeonTensorHandleTests.cpp | 427 OutputTensors outputTensors; variable
604 OutputTensors outputTensors; variable
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | cpp_c10d_extension.cpp | 51 std::vector<std::vector<at::Tensor>>& outputTensors, in allgather()
70 std::vector<std::vector<at::Tensor>>& outputTensors, in gather()
77 std::vector<at::Tensor>& outputTensors, in scatter()
84 std::vector<at::Tensor>& outputTensors, in reduce_scatter()
|
| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | Runtime.cpp | 98 …<ImportedOutputId> IRuntime::ImportOutputs(NetworkId networkId, const OutputTensors& outputTensors, in ImportOutputs()
115 const OutputTensors& outputTensors, in EnqueueWorkload()
125 const OutputTensors& outputTensors, in Execute()
629 …portedOutputId> RuntimeImpl::ImportOutputs(NetworkId networkId, const OutputTensors& outputTensors, in ImportOutputs()
646 const OutputTensors& outputTensors, in EnqueueWorkload()
698 const OutputTensors& outputTensors, in Execute()
|
| H A D | Threadpool.cpp | 71 const OutputTensors& outputTensors, in Schedule()
182 auto outputTensors = std::get<2>(*currentExecInProgress); in ProcessExecPriorities() local
|
| H A D | LoadedNetwork.cpp | 793 WorkloadData(const InputTensors& inputTensors, const OutputTensors& outputTensors) in WorkloadData()
840 const OutputTensors& outputTensors, in EnqueueWorkload()
1405 …nst armnn::Tensor GetOutputTensor(const LayerBindingId layerId, const OutputTensors& outputTensors) in GetOutputTensor()
1560 std::vector<ImportedOutputId> LoadedNetwork::ImportOutputs(const OutputTensors& outputTensors, in ImportOutputs()
1728 const OutputTensors& outputTensors, in Execute()
|
| /aosp_15_r20/external/armnn/src/backends/cl/test/ |
| H A D | ClImportTensorHandleTests.cpp | 181 OutputTensors outputTensors variable
390 OutputTensors outputTensors variable
532 OutputTensors outputTensors variable
680 OutputTensors outputTensors variable
802 OutputTensors outputTensors variable
965 OutputTensors outputTensors variable
1189 OutputTensors outputTensors variable
|
| H A D | ClFallbackTests.cpp | 132 OutputTensors outputTensors variable
257 OutputTensors outputTensors variable
414 OutputTensors outputTensors variable
551 OutputTensors outputTensors variable
|
| /aosp_15_r20/external/armnn/src/backends/reference/test/ |
| H A D | RefEndToEndTests.cpp | 158 armnn::OutputTensors outputTensors variable
227 OutputTensors outputTensors variable
322 OutputTensors outputTensors variable
389 OutputTensors outputTensors variable
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | ArmnnPreparedModel.cpp | 332 auto outputTensors = std::make_shared<armnn::OutputTensors>(); in execute() local
378 armnn::OutputTensors& outputTensors, in ExecuteGraph()
535 auto outputTensors = std::make_shared<armnn::OutputTensors>(); in executeFenced() local
667 armnn::OutputTensors outputTensors; in ExecuteWithDummyInputs() local
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | ArmnnPreparedModel.cpp | 317 armnn::OutputTensors& outputTensors, in ExecuteGraph()
415 armnn::OutputTensors outputTensors; in ExecuteWithDummyInputs() local
475 std::shared_ptr<armnn::OutputTensors>& outputTensors, in ScheduleGraphForExecution()
|
| H A D | ArmnnPreparedModel_1_2.cpp | 527 armnn::OutputTensors& outputTensors, in ExecuteGraph()
649 armnn::OutputTensors outputTensors; in ExecuteWithDummyInputs() local
705 auto outputTensors = std::make_shared<armnn::OutputTensors>(); in Execute() local
768 std::shared_ptr<armnn::OutputTensors>& outputTensors, in ScheduleGraphForExecution()
|
| H A D | ArmnnPreparedModel_1_3.cpp | 829 armnn::OutputTensors& outputTensors, in ExecuteGraph()
934 std::shared_ptr<armnn::OutputTensors>& outputTensors, in ScheduleGraphForExecution()
987 armnn::OutputTensors outputTensors; in ExecuteWithDummyInputs() local
1045 auto outputTensors = std::make_shared<armnn::OutputTensors>(); in Execute() local
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/ |
| H A D | ParserFlatbuffersFixture.hpp | 352 armnn::OutputTensors outputTensors; in RunTest() local
421 armnn::OutputTensors outputTensors; in RunTest() local
473 armnn::OutputTensors outputTensors; in RunTest() local
|
| /aosp_15_r20/external/armnn/src/profiling/test/ |
| H A D | FileOnlyProfilingDecoratorTests.cpp | 102 OutputTensors outputTensors variable
227 OutputTensors outputTensors variable
|
| /aosp_15_r20/external/armnn/tests/ExecuteNetwork/ |
| H A D | ArmNNExecutor.cpp | 436 armnn::OutputTensors outputTensors; in SetupInputsAndOutputs() local
619 void ArmNNExecutor::PrintOutputTensors(const armnn::OutputTensors* outputTensors, in PrintOutputTensors()
711 for (const auto& outputTensors: m_OutputTensorsVec) in CompareAndPrintResult() local
|
| /aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
| H A D | FoldPadTests.cpp | 607 OutputTensors outputTensors{{0, Tensor(outputInfo, optimizedData.data())}}; variable
731 OutputTensors outputTensors{{0, Tensor(outputInfo, optimizedData.data())}}; variable
856 OutputTensors outputTensors{{0, Tensor(outputInfo, optimizedData.data())}}; variable
|