Home
last modified time | relevance | path

Searched defs:outputShapes (Results 1 – 25 of 52) sorted by relevance

123

/aosp_15_r20/external/armnn/src/armnn/test/
H A DInferOutputTests.hpp21 std::vector<armnn::TensorShape>& outputShapes) in ArgMinMaxInferOutputShapeImpl()
39 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape4dTest() local
58 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape3dTest() local
77 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape2dTest() local
96 std::vector<armnn::TensorShape> outputShapes; in ArgMinMaxInferOutputShape1dTest() local
150 std::vector<armnn::TensorShape>& outputShapes) in PreluInferOutputShapeImpl()
170 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeSameDimsTest() local
190 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeInputBiggerTest() local
210 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeAlphaBiggerTest() local
230 std::vector<armnn::TensorShape> outputShapes; in PreluInferOutputShapeNoMatchTest() local
[all …]
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DSplitterEndToEndTestImpl.hpp24 const std::vector<TensorShape>& outputShapes, in CreateSplitterNetwork()
84 const std::vector<TensorShape> outputShapes{{ 2 }, { 2 }}; in Splitter1dEndToEnd() local
112 const std::vector<TensorShape> outputShapes{{ 2, 3 }, { 2, 3 }}; in Splitter2dDim0EndToEnd() local
147 const std::vector<TensorShape> outputShapes{{ 4, 1 }, { 4, 1 }, { 4, 1 }}; in Splitter2dDim1EndToEnd() local
185 const std::vector<TensorShape> outputShapes{{ 1, 4, 3 }, { 1, 4, 3 }}; in Splitter3dDim0EndToEnd() local
233 const std::vector<TensorShape> outputShapes{{ 2, 2, 3 }, { 2, 2, 3 }}; in Splitter3dDim1EndToEnd() local
281 const std::vector<TensorShape> outputShapes{{ 2, 4, 1 }, { 2, 4, 1 }, { 2, 4, 1 }}; in Splitter3dDim2EndToEnd() local
321 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim0EndToEnd() local
401 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim1EndToEnd() local
481 const std::vector<TensorShape> outputShapes{{ 2, 3, 2, 2 }, { 2, 3, 2, 2 }}; in Splitter4dDim2EndToEnd() local
[all …]
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DSplit.cpp33 const std::vector<Shape>& outputShapes) { in splitGeneric()
59 const std::vector<Shape>& outputShapes) { in splitFloat16()
66 const std::vector<Shape>& outputShapes) { in splitFloat32()
73 const std::vector<Shape>& outputShapes) { in splitQuant8()
80 const std::vector<Shape>& outputShapes) { in splitQuant8Signed()
87 const std::vector<Shape>& outputShapes) { in splitInt32()
/aosp_15_r20/external/android-nn-driver/
H A DArmnnPreparedModel_1_3.cpp68 std::vector<V1_2::OutputShape> outputShapes, in NotifyCallbackAndCheck()
83 std::vector<V1_2::OutputShape> outputShapes, in NotifyCallbackAndCheck()
320 std::vector<V1_2::OutputShape> outputShapes, in execute_1_2()
345 std::vector<V1_2::OutputShape> outputShapes, in execute_1_3()
348 { in execute_1_3()
550 std::vector<V1_2::OutputShape> &outputShapes, in PrepareMemoryForOutputs()
646 std::vector<V1_2::OutputShape> outputShapes(request.outputs.size()); in PrepareMemoryForIO() local
741 { in executeSynchronously()
788 { in executeSynchronously_1_3()
838 std::vector<V1_2::OutputShape> outputShapes(outputTensors.size()); in ExecuteGraph() local
[all …]
H A DArmnnPreparedModel_1_2.cpp63 std::vector<V1_2::OutputShape> outputShapes, in NotifyCallbackAndCheck()
295 std::vector<V1_2::OutputShape> outputShapes, in execute_1_2()
298 { in execute_1_2()
346 std::vector<V1_2::OutputShape> &outputShapes, in PrepareMemoryForOutputs()
430 std::vector<V1_2::OutputShape> outputShapes(request.outputs.size()); in PrepareMemoryForIO() local
493 { in executeSynchronously()
538 std::vector<V1_2::OutputShape> outputShapes(outputTensors.size()); in ExecuteGraph() local
776 std::vector<V1_2::OutputShape> outputShapes(outputTensorSize); in ScheduleGraphForExecution() local
/aosp_15_r20/hardware/interfaces/neuralnetworks/1.3/utils/src/
H A DCallbacks.cpp65 convertExecutionGeneralResultsHelper(const hidl_vec<V1_2::OutputShape>& outputShapes, in convertExecutionGeneralResultsHelper()
85 ErrorStatus status, const hidl_vec<V1_2::OutputShape>& outputShapes, in executionCallback()
131 const hidl_vec<V1_2::OutputShape>& outputShapes, in notify_1_2()
138 const hidl_vec<V1_2::OutputShape>& outputShapes, in notify_1_3()
/aosp_15_r20/external/armnn/delegate/test/
H A DSplitTest.cpp25 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitUint8Test() local
55 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitFp32Test() local
140 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitVUint8Test() local
178 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 }; in SplitVFp32Test() local
/aosp_15_r20/hardware/interfaces/neuralnetworks/1.2/utils/src/
H A DCallbacks.cpp56 convertExecutionGeneralResultsHelper(const hidl_vec<OutputShape>& outputShapes, in convertExecutionGeneralResultsHelper()
70 V1_0::ErrorStatus status, const hidl_vec<OutputShape>& outputShapes, const Timing& timing) { in executionCallback()
109 const hidl_vec<OutputShape>& outputShapes, in notify_1_2()
H A DBurstUtils.cpp142 const std::vector<V1_2::OutputShape>& outputShapes, in serialize()
355 std::vector<V1_2::OutputShape> outputShapes; in deserialize() local
589 const std::vector<V1_2::OutputShape>& outputShapes, in send()
/aosp_15_r20/hardware/interfaces/neuralnetworks/1.3/vts/functional/
H A DCallbacks.cpp92 const hidl_vec<OutputShape>& outputShapes, in notify_1_2()
98 const hidl_vec<OutputShape>& outputShapes, in notify_1_3()
124 hidl_vec<OutputShape> outputShapes, Timing timing) { in notifyInternal()
H A DValidateRequest.cpp75 const auto& outputShapes = executionCallback->getOutputShapes(); in validate() local
89 const Timing& timing) { in validate()
184 [](ErrorStatus error, const hidl_vec<OutputShape>& outputShapes, const Timing& timing) { in validateRequestFailure()
H A DQualityOfServiceTests.cpp186 hidl_vec<OutputShape> outputShapes = callback->getOutputShapes(); in executeAsynchronously() local
202 const Timing& timing) { in executeSynchronously()
/aosp_15_r20/hardware/interfaces/neuralnetworks/1.2/vts/functional/
H A DValidateRequest.cpp71 const auto& outputShapes = executionCallback->getOutputShapes(); in validate() local
85 const Timing& timing) { in validate()
162 [](ErrorStatus error, const hidl_vec<OutputShape>& outputShapes, const Timing& timing) { in validateRequestFailure()
H A DCallbacks.cpp82 const hidl_vec<OutputShape>& outputShapes, in notify_1_2()
125 const hidl_vec<OutputShape>& outputShapes, in notifyInternal()
H A DGeneratedTestHarness.cpp190 hidl_vec<OutputShape>* outputShapes, in ExecutePreparedModel()
227 hidl_vec<OutputShape> outputShapes; in EvaluatePreparedModel() local
/aosp_15_r20/out/soong/.intermediates/hardware/interfaces/neuralnetworks/1.3/[email protected]_genc++/gen/android/hardware/neuralnetworks/1.3/
DExecutionCallbackAll.cpp153 …ce *_hidl_this, ::android::hardware::details::HidlInstrumentor *_hidl_this_instrumentor, ::android… in _hidl_notify_1_3()
249 …::V1_0::ErrorStatus status, const ::android::hardware::hidl_vec<::android::hardware::neuralnetwork… in notify_1_2()
257 …::V1_3::ErrorStatus status, const ::android::hardware::hidl_vec<::android::hardware::neuralnetwork… in notify_1_3()
365 …:android::hardware::hidl_vec<::android::hardware::neuralnetworks::V1_2::OutputShape>* outputShapes; in _hidl_notify_1_3() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/
DExecutionBuilder.cpp724 std::vector<OutputShape> outputShapes = getInitialOutputShapes(); in computeInternal() local
1114 [this, mode](ErrorStatus error, const std::vector<OutputShape>& outputShapes) { in compute()
1136 std::vector<OutputShape> outputShapes(mOutputs.size()); in getInitialOutputShapes() local
1167 const std::vector<OutputShape>& outputShapes) { in updateOutputShapes()
1197 int ExecutionBuilder::finishComputation(int result, const std::vector<OutputShape>& outputShapes, in finishComputation()
1516 std::vector<OutputShape> outputShapes; in compute() local
DExecutionCallback.cpp29 void ExecutionCallback::notify(ErrorStatus status, const std::vector<OutputShape>& outputShapes, in notify()
115 std::vector<OutputShape> outputShapes, Timing timing) { in notifyInternal()
/aosp_15_r20/out/soong/.intermediates/hardware/interfaces/neuralnetworks/1.2/[email protected]_genc++/gen/android/hardware/neuralnetworks/1.2/
DExecutionCallbackAll.cpp142 …ce *_hidl_this, ::android::hardware::details::HidlInstrumentor *_hidl_this_instrumentor, ::android… in _hidl_notify_1_2()
238 …::V1_0::ErrorStatus status, const ::android::hardware::hidl_vec<::android::hardware::neuralnetwork… in notify_1_2()
346 …:android::hardware::hidl_vec<::android::hardware::neuralnetworks::V1_2::OutputShape>* outputShapes; in _hidl_notify_1_2() local
/aosp_15_r20/packages/modules/NeuralNetworks/driver/sample_hidl/
DSampleDriverUtils.cpp81 const hardware::hidl_vec<V1_2::OutputShape>& outputShapes, V1_2::Timing timing) { in notify()
89 const hardware::hidl_vec<V1_2::OutputShape>& outputShapes, V1_2::Timing timing) { in notify()
DSampleDriver.cpp411 const hardware::hidl_vec<V1_2::OutputShape>& outputShapes) { in updateDeviceMemories()
482 hardware::hidl_vec<V1_2::OutputShape> outputShapes = convertToV1_2(executor.getOutputShapes()); in asyncExecute() local
614 hardware::hidl_vec<V1_2::OutputShape> outputShapes = convertToV1_2(executor.getOutputShapes()); in executeSynchronouslyBase() local
831 hardware::hidl_vec<V1_2::OutputShape> outputShapes = in execute() local
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_ops.cpp27 std::vector<std::vector<ExprHandle>> outputShapes = {{N}, {M}, {}}; in TEST() local
57 std::vector<std::vector<ExprHandle>> outputShapes = { in TEST() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DArmnnPreparedModel.cpp189 std::vector<OutputShape> &outputShapes, in PrepareMemoryForOutputs()
273 std::vector<OutputShape> outputShapes(request.outputs.size()); in PrepareMemoryForIO() local
353 std::vector<OutputShape> outputShapes(outputTensors->size()); in execute() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DExecutionBurstServer.cpp89 const V1_2::Timing& timing) { in execute()
115 const std::vector<V1_2::OutputShape>& outputShapes, in serialize()
477 const std::vector<V1_2::OutputShape>& outputShapes, in send()
/aosp_15_r20/packages/modules/NeuralNetworks/driver/sample/
DCanonicalPreparedModel.cpp74 const std::vector<OutputShape>& outputShapes) { in updateDeviceMemories()
167 const auto& outputShapes = executor.getOutputShapes(); in execute() local

123