| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | ProcessGroup.hpp | 217 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()
|
| D | PyProcessGroup.hpp | 71 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()
|
| D | ProcessGroupWrapper.cpp | 433 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()
|
| D | ProcessGroupMPI.cpp | 124 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()
|
| D | ProcessGroupUCC.cpp | 771 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()
|
| /external/pytorch/test/cpp_extensions/ |
| D | cpp_c10d_extension.cpp | 52 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()
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | EndToEndTestImpl.hpp | 68 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 …]
|
| D | StridedSliceAsyncEndToEndTest.hpp | 57 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
|
| /external/armnn/src/backends/reference/test/ |
| D | RefEndToEndTests.cpp | 154 armnn::InputTensors inputTensors variable
222 InputTensors inputTensors variable
318 InputTensors inputTensors variable
384 InputTensors inputTensors variable
|
| /external/armnn/src/armnnTfLiteParser/test/ |
| D | FullyConnected.cpp | 23 std::string inputTensors = "[ 0, 2 ]"; in FullyConnectedFixture() local
204 std::string inputTensors = "[ 0, 1 ]"; in FullyConnectedNonConstWeightsFixture() local
|
| D | ParserFlatbuffersFixture.hpp | 293 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
|
| D | DepthwiseConvolution2D.cpp | 22 std::string inputTensors = "[ 0, 2 ]"; in DepthwiseConvolution2dFixture() local
238 std::string inputTensors = "[ 0, 2 ]"; in DepthwiseConvolution2dFixture2() local
|
| /external/armnn/src/armnn/ |
| D | Runtime.cpp | 92 …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()
|
| D | Threadpool.cpp | 70 const InputTensors& inputTensors, in Schedule()
181 auto inputTensors = std::get<1>(*currentExecInProgress); in ProcessExecPriorities() local
|
| /external/armnn/src/profiling/test/ |
| D | FileOnlyProfilingDecoratorTests.cpp | 98 InputTensors inputTensors variable
223 InputTensors inputTensors variable
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonFallbackTests.cpp | 118 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
|
| /external/armnn/src/backends/cl/test/ |
| D | ClImportTensorHandleTests.cpp | 177 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
|
| D | ClFallbackTests.cpp | 126 InputTensors inputTensors variable
251 InputTensors inputTensors variable
408 InputTensors inputTensors variable
545 InputTensors inputTensors variable
|
| D | Fp16SupportTest.cpp | 93 InputTensors inputTensors variable
|
| /external/android-nn-driver/ |
| D | ArmnnPreparedModel.cpp | 316 armnn::InputTensors& inputTensors, in ExecuteGraph()
401 armnn::InputTensors inputTensors; in ExecuteWithDummyInputs() local
474 std::shared_ptr<armnn::InputTensors>& inputTensors, in ScheduleGraphForExecution()
|
| /external/armnn/shim/sl/canonical/ |
| D | ArmnnPreparedModel.cpp | 331 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
|
| /external/armnn/samples/ |
| D | DynamicSample.cpp | 67 InputTensors inputTensors in main() local
|
| D | SimpleSample.cpp | 80 InputTensors inputTensors{{0, armnn::ConstTensor(inputTensorInfo, in main() local
|
| /external/armnn/delegate/opaque/src/ |
| D | Redefine.hpp | 29 const int* inputTensors; in VisitCastOperator() local
|
| /external/armnn/src/armnnUtils/ |
| D | TensorIOUtils.hpp | 20 armnn::InputTensors inputTensors; in MakeInputTensors() local
|