| /frameworks/ml/nn/common/operations/ |
| D | Split.cpp | 32 const std::vector<Shape>& outputShapes) { in splitGeneric() argument
47 const int copySize = outputShapes[i].dimensions[axis] * baseInnerSize; in splitGeneric()
58 const std::vector<Shape>& outputShapes) { in splitFloat16() argument
60 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
65 const std::vector<Shape>& outputShapes) { in splitFloat32() argument
67 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
72 const std::vector<Shape>& outputShapes) { in splitQuant8() argument
74 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
79 const std::vector<Shape>& outputShapes) { in splitQuant8Signed() argument
81 return splitGeneric<int8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8Signed()
[all …]
|
| /frameworks/ml/nn/runtime/ |
| D | Callbacks.cpp | 102 const hidl_vec<OutputShape>& outputShapes, in notify_1_2() argument
104 return notifyInternal(false, static_cast<ErrorStatus>(errorStatus), outputShapes, timing); in notify_1_2()
108 const hidl_vec<OutputShape>& outputShapes, in notify_1_3() argument
110 return notifyInternal(false, errorStatus, outputShapes, timing); in notify_1_3()
203 std::vector<OutputShape> outputShapes, in notifyInternal() argument
209 if (outputShapes.size() == 0) { in notifyInternal()
213 outputShapes = {}; in notifyInternal()
219 if (outputShapes.size() != 0) { in notifyInternal()
223 outputShapes = {}; in notifyInternal()
240 mOutputShapes = std::move(outputShapes); in notifyInternal()
|
| D | ExecutionBuilder.cpp | 493 auto [n2, outputShapes, timing] = executor->computeOnCpuFallback(); in cpuFallbackPartial()
494 return {n2, std::move(outputShapes), timing, executor}; in cpuFallbackPartial()
506 std::vector<OutputShape> outputShapes = executionBuilder->getInitialOutputShapes(); in asyncStartComputePartitioned() local
531 executionCallback->notify(ErrorStatus::NONE, outputShapes, timing); in asyncStartComputePartitioned()
540 if (!executor->updateOutputShapes(stepOutputShapes, &outputShapes)) { in asyncStartComputePartitioned()
556 executionCallback->notify(stepStatus, outputShapes, kNoTiming); in asyncStartComputePartitioned()
583 !fallbackExecutor->updateOutputShapes(fallbackOutputShapes, &outputShapes)) { in asyncStartComputePartitioned()
599 executionCallback->notify(fallbackStatus, outputShapes, kNoTiming); in asyncStartComputePartitioned()
821 auto wrappedFinish = [this](ErrorStatus error, const std::vector<OutputShape>& outputShapes) { in compute() argument
822 return finishWithoutSyncFence(error, outputShapes); in compute()
[all …]
|
| D | Callbacks.h | 283 const hal::hidl_vec<hal::OutputShape>& outputShapes,
322 const hal::hidl_vec<hal::OutputShape>& outputShapes,
327 const hal::hidl_vec<hal::OutputShape>& outputShapes, in notify() argument
329 return notify_1_3(status, outputShapes, timing); in notify()
465 std::vector<hal::OutputShape> outputShapes,
|
| D | Manager.cpp | 374 std::vector<OutputShape> outputShapes; in execute() local
393 std::tie(n, outputShapes, timing, burstFallback) = in execute()
403 std::tie(n, outputShapes, timing) = mPreparedModel->execute( in execute()
409 return {n, std::move(outputShapes), timing}; in execute()
426 return {ANEURALNETWORKS_NO_ERROR, std::move(outputShapes), timing}; in execute()
707 const auto& outputShapes = executor.getOutputShapes(); in computeOnCpu() local
708 return {err, outputShapes, kNoTiming}; in computeOnCpu()
740 const auto [result, outputShapes, timing] = execute(inputs, outputs, memories, nullptr, measure, in executeFenced()
|
| D | ExecutionBuilder.h | 107 const std::vector<hal::OutputShape>& outputShapes);
137 bool updateOutputShapes(const std::vector<hal::OutputShape>& outputShapes);
|
| D | VersionedInterfaces.cpp | 316 [&result](ErrorStatus error, const hidl_vec<OutputShape>& outputShapes, in executeSynchronously()
318 result = getExecutionResult(error, outputShapes, timing); in executeSynchronously()
343 [&result](V1_0::ErrorStatus error, const hidl_vec<OutputShape>& outputShapes, in executeSynchronously()
345 result = getExecutionResult(convertToV1_3(error), outputShapes, timing); in executeSynchronously()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestVersionedInterfaces.cpp | 366 const std::vector<OutputShape>& outputShapes, const Timing& timing) { in makeExecute_1_2Return() argument
367 return [launchStatus, returnStatus, outputShapes, timing]( in makeExecute_1_2Return()
370 cb->notify_1_2(returnStatus, outputShapes, timing); in makeExecute_1_2Return()
375 const std::vector<OutputShape>& outputShapes, const Timing& timing) { in makeExecute_1_3Return() argument
376 return [launchStatus, returnStatus, outputShapes, timing]( in makeExecute_1_3Return()
381 cb->notify_1_3(returnStatus, outputShapes, timing); in makeExecute_1_3Return()
386 const std::vector<OutputShape>& outputShapes, in makeExecuteSynchronouslyReturn() argument
388 return [status, outputShapes, timing](const V1_0::Request& /*request*/, in makeExecuteSynchronouslyReturn()
391 cb(status, outputShapes, timing); in makeExecuteSynchronouslyReturn()
396 const std::vector<OutputShape>& outputShapes, in makeExecuteSynchronously_1_3Return() argument
[all …]
|
| D | TestIntrospectionControl.cpp | 518 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in executeSynchronously_1_3() argument
519 cb(convertToV1_3(status), outputShapes, timing); in executeSynchronously_1_3()
|
| /frameworks/ml/nn/driver/sample/ |
| D | SampleDriverUtils.cpp | 63 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in notify() argument
64 const auto ret = callback->notify_1_2(convertToV1_0(status), outputShapes, timing); in notify()
71 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in notify() argument
72 const auto ret = callback->notify_1_3(status, outputShapes, timing); in notify()
|
| D | SampleDriver.cpp | 411 const hidl_vec<OutputShape>& outputShapes) { in updateDeviceMemories() argument
417 if (!bufferWrappers[poolIndex]->updateDimensions(outputShapes[i].dimensions)) { in updateDeviceMemories()
437 if (!outputShapes[i].isSufficient) { in updateDeviceMemories()
439 << ": actual shape = " << toString(outputShapes[i].dimensions); in updateDeviceMemories()
482 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in asyncExecute() local
486 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in asyncExecute()
497 notify(callback, executionStatus, outputShapes, timing); in asyncExecute()
499 notify(callback, executionStatus, outputShapes, kNoTiming); in asyncExecute()
607 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in executeSynchronouslyBase() local
611 updateDeviceMemories(executionStatus, request, bufferWrappers, outputShapes); in executeSynchronouslyBase()
[all …]
|
| D | SampleDriverUtils.h | 41 const hal::hidl_vec<hal::OutputShape>& outputShapes, hal::Timing timing);
44 const hal::hidl_vec<hal::OutputShape>& outputShapes, hal::Timing timing);
|
| /frameworks/ml/nn/common/ |
| D | ExecutionBurstServer.cpp | 87 V1_0::ErrorStatus status, const hidl_vec<OutputShape>& outputShapes, in execute()
90 returnedOutputShapes = outputShapes; in execute()
113 const std::vector<OutputShape>& outputShapes, Timing timing) { in serialize() argument
115 size_t count = 2 + outputShapes.size(); in serialize()
116 for (const auto& outputShape : outputShapes) { in serialize()
129 /*.numberOfOperands=*/static_cast<uint32_t>(outputShapes.size())}); in serialize()
134 for (const auto& operand : outputShapes) { in serialize()
466 const std::vector<OutputShape>& outputShapes, Timing timing) { in send() argument
467 const std::vector<FmqResultDatum> serialized = serialize(errorStatus, outputShapes, timing); in send()
636 const auto [errorStatus, outputShapes, returnedTiming] = in task()
[all …]
|
| D | ExecutionBurstController.cpp | 156 std::vector<OutputShape> outputShapes; in deserialize() local
211 outputShapes.push_back({/*.dimensions=*/dimensions, /*.isSufficient=*/isSufficient}); in deserialize()
231 return std::make_tuple(errorStatus, std::move(outputShapes), timing); in deserialize()
576 V1_0::ErrorStatus status, std::vector<OutputShape> outputShapes, Timing timing, in getExecutionResult() argument
579 getExecutionResult(convertToV1_3(status), std::move(outputShapes), timing); in getExecutionResult()
616 auto [status, outputShapes, timing] = std::move(*result); in compute()
617 return getExecutionResult(status, std::move(outputShapes), timing, /*fallback=*/false); in compute()
|
| D | CpuExecutor.cpp | 1437 std::vector<Shape> outputShapes(numOutputs); in executeOperation() local
1439 outputShapes[i] = operands[outs[i]].shape(); in executeOperation()
1442 success = splitPrepare(input.shape(), axis, numOutputs, &outputShapes); in executeOperation()
1445 outputShapes[i], &result); in executeOperation()
1455 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1464 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1473 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1482 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1491 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
|
| D | Utils.cpp | 1953 ErrorStatus status, std::vector<OutputShape> outputShapes, Timing timing) { in getExecutionResult() argument
1958 !outputShapes.empty()) { in getExecutionResult()
1960 outputShapes.clear(); in getExecutionResult()
1966 return {n, std::move(outputShapes), timing}; in getExecutionResult()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 126 const std::vector<Shape>& outputShapes);
130 const std::vector<Shape>& outputShapes);
134 const std::vector<Shape>& outputShapes);
138 const std::vector<Shape>& outputShapes);
142 const std::vector<Shape>& outputShapes);
|
| D | ExecutionBurstServer.h | 56 const std::vector<hardware::neuralnetworks::V1_2::OutputShape>& outputShapes,
164 const std::vector<hardware::neuralnetworks::V1_2::OutputShape>& outputShapes,
|
| D | Utils.h | 427 hal::ErrorStatus status, std::vector<hal::OutputShape> outputShapes, hal::Timing timing);
|