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()
171 "IndexedShapeWrapper.cpp",294 "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()
41 IndexedShapeWrapper aShapeIndexed(aShape); in compute()42 IndexedShapeWrapper bShapeIndexed(bShape); in compute()43 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
57 const IndexedShapeWrapper indexedOutput = IndexedShapeWrapper(outputShape); in evalGeneric()58 const IndexedShapeWrapper indexedInput = IndexedShapeWrapper(inputShape); in evalGeneric()
49 IndexedShapeWrapper aShapeIndexed(aShape); in eval()50 IndexedShapeWrapper bShapeIndexed(bShape); in eval()51 IndexedShapeWrapper outputShapeIndexed(outputShape); in eval()
43 IndexedShapeWrapper aShapeIndexed(aShape); in compute()44 IndexedShapeWrapper bShapeIndexed(bShape); in compute()45 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
215 IndexedShapeWrapper aShapeIndexed(aShape); in executeInt32()216 IndexedShapeWrapper bShapeIndexed(bShape); in executeInt32()217 IndexedShapeWrapper outputShapeIndexed(outputShape); in executeInt32()
30 class IndexedShapeWrapper {32 IndexedShapeWrapper(const Shape& wrapped_shape);