MCPcopy Create free account
hub / github.com/alibaba/MNN / _hasZeroShapeOutput

Function _hasZeroShapeOutput

source/geometry/GeometryComputerUtils.cpp:20–29  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

18#define DEFAULT_ALLOCATE_SIZE 32
19namespace MNN {
20static bool _hasZeroShapeOutput(const Schedule::OpCacheInfo& info) {
21 for (auto t : info.outputs) {
22 for (int v = 0; v < t->dimensions(); ++v) {
23 if (t->length(v) <= 0) {
24 return true;
25 }
26 }
27 }
28 return false;
29}
30flatbuffers::Offset<Op> GeometryComputerUtils::makePool(flatbuffers::FlatBufferBuilder& builder, std::pair<int, int> kernel, std::pair<int, int> stride, PoolType type, MNN::PoolPadType pad, std::pair<int, int> pads, bool isglobal, AvgPoolCountType countType) {
31 PoolBuilder poolB(builder);
32 poolB.add_type(type);

Callers 1

Calls 2

dimensionsMethod · 0.45
lengthMethod · 0.45

Tested by

no test coverage detected