Searched refs:IndexedShapeWrapper (Results 1 – 10 of 10) sorted by relevance
28 IndexedShapeWrapper::IndexedShapeWrapper(const Shape& wrapped_shape) : shape(&wrapped_shape) { in IndexedShapeWrapper() function in android::nn::IndexedShapeWrapper36 bool IndexedShapeWrapper::nextIndexInplace(std::vector<uint32_t>* index, bool* lastIndex) const { in nextIndexInplace()61 bool IndexedShapeWrapper::indexToFlatIndex(const std::vector<uint32_t>& index, in indexToFlatIndex()72 bool IndexedShapeWrapper::broadcastedIndexToFlatIndex(const std::vector<uint32_t>& index, in broadcastedIndexToFlatIndex()88 bool IndexedShapeWrapper::isValid(const std::vector<uint32_t>& index) const { in isValid()
139 "IndexedShapeWrapper.cpp",285 "IndexedShapeWrapper.cpp",
37 IndexedShapeWrapper aShapeIndexed(aShape); in evalGeneric()38 IndexedShapeWrapper bShapeIndexed(bShape); in evalGeneric()39 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()63 IndexedShapeWrapper aShapeIndexed(aShape); in evalQuant8()64 IndexedShapeWrapper bShapeIndexed(bShape); in evalQuant8()65 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalQuant8()
36 IndexedShapeWrapper baseShapeIndexed(baseShape); in evalGeneric()37 IndexedShapeWrapper exponentShapeIndexed(exponentShape); in evalGeneric()38 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()
35 IndexedShapeWrapper aShapeIndexed(aShape); in compute()36 IndexedShapeWrapper bShapeIndexed(bShape); in compute()37 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
49 const IndexedShapeWrapper indexedOutput = IndexedShapeWrapper(outputShape); in evalGeneric()50 const IndexedShapeWrapper indexedInput = IndexedShapeWrapper(inputShape); in evalGeneric()
48 IndexedShapeWrapper aShapeIndexed(aShape); in eval()49 IndexedShapeWrapper bShapeIndexed(bShape); in eval()50 IndexedShapeWrapper outputShapeIndexed(outputShape); in eval()
37 IndexedShapeWrapper aShapeIndexed(aShape); in compute()38 IndexedShapeWrapper bShapeIndexed(bShape); in compute()39 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
215 IndexedShapeWrapper aShapeIndexed(aShape); in executeInt32()216 IndexedShapeWrapper bShapeIndexed(bShape); in executeInt32()217 IndexedShapeWrapper outputShapeIndexed(outputShape); in executeInt32()
32 class IndexedShapeWrapper {34 IndexedShapeWrapper(const Shape& wrapped_shape);