Home
last modified time | relevance | path

Searched refs:outputShapes (Results 1 – 13 of 13) sorted by relevance

/frameworks/ml/nn/common/operations/
DSplit.cpp30 const std::vector<Shape>& outputShapes) { in splitGeneric() argument
45 const int copySize = outputShapes[i].dimensions[axis] * baseInnerSize; in splitGeneric()
56 const std::vector<Shape>& outputShapes) { in splitFloat16() argument
58 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
63 const std::vector<Shape>& outputShapes) { in splitFloat32() argument
65 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
70 const std::vector<Shape>& outputShapes) { in splitQuant8() argument
72 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
77 const std::vector<Shape>& outputShapes) { in splitInt32() argument
79 return splitGeneric<int32_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitInt32()
/frameworks/ml/nn/runtime/
DCallbacks.cpp78 const hidl_vec<OutputShape>& outputShapes, in notify_1_2() argument
82 if (outputShapes.size() == 0) { in notify_1_2()
89 if (outputShapes.size() != 0) { in notify_1_2()
96 notifyInternal(errorStatus, outputShapes, timing); in notify_1_2()
181 const hidl_vec<OutputShape>& outputShapes, in notifyInternal() argument
192 mOutputShapes = outputShapes; in notifyInternal()
DExecutionBuilder.cpp413 std::vector<OutputShape>* outputShapes) { in cpuFallbackPartial() argument
430 if (!executor->updateOutputShapes(stepOutputShapes, outputShapes)) { in cpuFallbackPartial()
436 executionCallback->notify(status, *outputShapes, kNoTiming); in cpuFallbackPartial()
451 std::vector<OutputShape> outputShapes; in asyncStartComputePartitioned() local
453 executionBuilder->initializeOutputShapes(&outputShapes); in asyncStartComputePartitioned()
468 executionCallback->notify(ErrorStatus::NONE, outputShapes, timing); in asyncStartComputePartitioned()
477 &outputShapes)) { in asyncStartComputePartitioned()
493 if (!executor->updateOutputShapes(stepOutputShapes, &outputShapes)) { in asyncStartComputePartitioned()
505 &outputShapes)) { in asyncStartComputePartitioned()
514 executionCallback->notify(status, outputShapes, kNoTiming); in asyncStartComputePartitioned()
[all …]
DCallbacks.h239 Return<void> notify_1_2(ErrorStatus status, const hidl_vec<OutputShape>& outputShapes,
243 Return<void> notify(ErrorStatus status, const hidl_vec<OutputShape>& outputShapes, in notify() argument
245 return notify_1_2(status, outputShapes, timing); in notify()
363 void notifyInternal(ErrorStatus errorStatus, const hidl_vec<OutputShape>& outputShapes,
DExecutionBuilder.h99 void initializeOutputShapes(std::vector<OutputShape>* outputShapes) const;
111 ErrorStatus finish(ErrorStatus error, const std::vector<OutputShape>& outputShapes);
127 bool updateOutputShapes(const std::vector<OutputShape>& outputShapes);
DVersionedInterfaces.cpp187 [&result](ErrorStatus error, const hidl_vec<OutputShape>& outputShapes, in executeSynchronously()
189 result = std::make_tuple(error, outputShapes, timing); in executeSynchronously()
/frameworks/ml/nn/driver/sample/
DSampleDriver.cpp220 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in notify() argument
221 return callback->notify_1_2(status, outputShapes, timing); in notify()
248 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in asyncExecute() local
255 returned = notify(callback, executionStatus, outputShapes, timing); in asyncExecute()
257 returned = notify(callback, executionStatus, outputShapes, kNoTiming); in asyncExecute()
335 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in executeSynchronously() local
341 cb(executionStatus, outputShapes, timing); in executeSynchronously()
343 cb(executionStatus, outputShapes, kNoTiming); in executeSynchronously()
410 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in execute() local
417 return std::make_tuple(executionStatus, outputShapes, timing); in execute()
[all …]
/frameworks/ml/nn/common/
DExecutionBurstServer.cpp72 ErrorStatus status, const hidl_vec<OutputShape>& outputShapes, in execute()
75 returnedOutputShapes = outputShapes; in execute()
98 const std::vector<OutputShape>& outputShapes, Timing timing) { in serialize() argument
100 size_t count = 2 + outputShapes.size(); in serialize()
101 for (const auto& outputShape : outputShapes) { in serialize()
114 /*.numberOfOperands=*/static_cast<uint32_t>(outputShapes.size())}); in serialize()
119 for (const auto& operand : outputShapes) { in serialize()
407 const std::vector<OutputShape>& outputShapes, Timing timing) { in send() argument
408 const std::vector<FmqResultDatum> serialized = serialize(errorStatus, outputShapes, timing); in send()
577 const auto [errorStatus, outputShapes, returnedTiming] = in task()
[all …]
DExecutionBurstController.cpp142 std::vector<OutputShape> outputShapes; in deserialize() local
197 outputShapes.push_back({/*.dimensions=*/dimensions, /*.isSufficient=*/isSufficient}); in deserialize()
217 return std::make_tuple(errorStatus, std::move(outputShapes), timing); in deserialize()
538 auto [status, outputShapes, timing, fallback] = tryCompute(request, measure, memoryIds); in compute()
540 return {status, std::move(outputShapes), timing}; in compute()
569 auto [status, outputShapes, timing] = std::move(*result); in tryCompute()
570 return {status, std::move(outputShapes), timing, /*fallback=*/false}; in tryCompute()
DCpuExecutor.cpp1475 std::vector<Shape> outputShapes(numOutputs); in executeOperation() local
1477 outputShapes[i] = mOperands[outs[i]].shape(); in executeOperation()
1480 success = splitPrepare(input.shape(), axis, numOutputs, &outputShapes); in executeOperation()
1483 outputShapes[i], &result); in executeOperation()
1493 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1502 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1511 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1520 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
/frameworks/ml/nn/common/include/
DOperations.h127 const std::vector<Shape>& outputShapes);
131 const std::vector<Shape>& outputShapes);
135 const std::vector<Shape>& outputShapes);
139 const std::vector<Shape>& outputShapes);
DExecutionBurstServer.h49 const std::vector<OutputShape>& outputShapes, Timing timing);
145 bool send(ErrorStatus errorStatus, const std::vector<OutputShape>& outputShapes, Timing timing);
/frameworks/ml/nn/runtime/test/
DTestExecution.cpp102 [&cb](ErrorStatus error, const hidl_vec<OutputShape>& outputShapes, in executeSynchronously()
103 const Timing& timing) { cb(error, outputShapes, timing); }); in executeSynchronously() argument