Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DSplit.cpp32 const std::vector<Shape>& outputShapes) { in splitGeneric()
58 const std::vector<Shape>& outputShapes) { in splitFloat16()
65 const std::vector<Shape>& outputShapes) { in splitFloat32()
72 const std::vector<Shape>& outputShapes) { in splitQuant8()
79 const std::vector<Shape>& outputShapes) { in splitQuant8Signed()
86 const std::vector<Shape>& outputShapes) { in splitInt32()
/frameworks/ml/nn/runtime/
DCallbacks.cpp102 const hidl_vec<OutputShape>& outputShapes, in notify_1_2()
108 const hidl_vec<OutputShape>& outputShapes, in notify_1_3()
203 std::vector<OutputShape> outputShapes, in notifyInternal()
DCallbacks.h327 const hal::hidl_vec<hal::OutputShape>& outputShapes, in notify()
DExecutionBuilder.cpp506 std::vector<OutputShape> outputShapes = executionBuilder->getInitialOutputShapes(); in asyncStartComputePartitioned() local
821 auto wrappedFinish = [this](ErrorStatus error, const std::vector<OutputShape>& outputShapes) { in compute()
874 std::vector<OutputShape> outputShapes(mOutputs.size()); in getInitialOutputShapes() local
897 bool ExecutionBuilder::updateOutputShapes(const std::vector<OutputShape>& outputShapes) { in updateOutputShapes()
926 const std::vector<OutputShape>& outputShapes) { in finishWithoutSyncFence()
DManager.cpp374 std::vector<OutputShape> outputShapes; in execute() local
707 const auto& outputShapes = executor.getOutputShapes(); in computeOnCpu() local
DVersionedInterfaces.cpp317 const Timing& timing) { in executeSynchronously()
344 const Timing& timing) { in executeSynchronously()
/frameworks/ml/nn/driver/sample/
DSampleDriverUtils.cpp63 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in notify()
71 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in notify()
DSampleDriver.cpp411 const hidl_vec<OutputShape>& outputShapes) { in updateDeviceMemories()
482 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in asyncExecute() local
607 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in executeSynchronouslyBase() local
819 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in execute() local
/frameworks/ml/nn/common/
DExecutionBurstServer.cpp88 const Timing& timing) { in execute()
113 const std::vector<OutputShape>& outputShapes, Timing timing) { in serialize()
466 const std::vector<OutputShape>& outputShapes, Timing timing) { in send()
DExecutionBurstController.cpp156 std::vector<OutputShape> outputShapes; in deserialize() local
576 V1_0::ErrorStatus status, std::vector<OutputShape> outputShapes, Timing timing, in getExecutionResult()
DCpuExecutor.cpp1437 std::vector<Shape> outputShapes(numOutputs); in executeOperation() local
DUtils.cpp1953 ErrorStatus status, std::vector<OutputShape> outputShapes, Timing timing) { in getExecutionResult()
/frameworks/ml/nn/runtime/test/
DTestVersionedInterfaces.cpp366 const std::vector<OutputShape>& outputShapes, const Timing& timing) { in makeExecute_1_2Return()
375 const std::vector<OutputShape>& outputShapes, const Timing& timing) { in makeExecute_1_3Return()
386 const std::vector<OutputShape>& outputShapes, in makeExecuteSynchronouslyReturn()
396 const std::vector<OutputShape>& outputShapes, in makeExecuteSynchronously_1_3Return()
DTestIntrospectionControl.cpp518 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in executeSynchronously_1_3()