Searched defs:outputShapes (Results 1 – 10 of 10) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Split.cpp | 30 const std::vector<Shape>& outputShapes) { in splitGeneric()
56 const std::vector<Shape>& outputShapes) { in splitFloat16()
63 const std::vector<Shape>& outputShapes) { in splitFloat32()
70 const std::vector<Shape>& outputShapes) { in splitQuant8()
77 const std::vector<Shape>& outputShapes) { in splitInt32()
|
| /frameworks/ml/nn/runtime/ |
| D | Callbacks.cpp | 78 const hidl_vec<OutputShape>& outputShapes, in notify_1_2()
181 const hidl_vec<OutputShape>& outputShapes, in notifyInternal()
|
| D | Callbacks.h | 243 Return<void> notify(ErrorStatus status, const hidl_vec<OutputShape>& outputShapes, in notify()
|
| D | ExecutionBuilder.cpp | 413 std::vector<OutputShape>* outputShapes) { in cpuFallbackPartial()
451 std::vector<OutputShape> outputShapes; in asyncStartComputePartitioned() local
559 auto wrappedFinish = [this](ErrorStatus error, const std::vector<OutputShape>& outputShapes) { in compute()
624 bool ExecutionBuilder::updateOutputShapes(const std::vector<OutputShape>& outputShapes) { in updateOutputShapes()
640 ErrorStatus ExecutionBuilder::finish(ErrorStatus, const std::vector<OutputShape>& outputShapes) { in finish()
1002 const auto& outputShapes = executor.getOutputShapes(); in computeOnCpu() local
|
| D | VersionedInterfaces.cpp | 188 const Timing& timing) { in executeSynchronously()
|
| /frameworks/ml/nn/driver/sample/ |
| D | SampleDriver.cpp | 220 const hidl_vec<OutputShape>& outputShapes, Timing timing) { in notify()
248 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in asyncExecute() local
335 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in executeSynchronously() local
410 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in execute() local
|
| /frameworks/ml/nn/common/ |
| D | ExecutionBurstServer.cpp | 73 const Timing& timing) { in execute()
98 const std::vector<OutputShape>& outputShapes, Timing timing) { in serialize()
407 const std::vector<OutputShape>& outputShapes, Timing timing) { in send()
|
| D | ExecutionBurstController.cpp | 142 std::vector<OutputShape> outputShapes; in deserialize() local
|
| D | CpuExecutor.cpp | 1475 std::vector<Shape> outputShapes(numOutputs); in executeOperation() local
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestExecution.cpp | 103 const Timing& timing) { cb(error, outputShapes, timing); }); in executeSynchronously()
|