Searched refs:IndexedShapeWrapper (Results 1 – 10 of 10) sorted by relevance
24 IndexedShapeWrapper::IndexedShapeWrapper(const Shape& wrapped_shape) : shape(&wrapped_shape) { in IndexedShapeWrapper() function in android::nn::IndexedShapeWrapper32 bool IndexedShapeWrapper::nextIndexInplace(std::vector<uint32_t>* index, bool* lastIndex) const { in nextIndexInplace()57 bool IndexedShapeWrapper::indexToFlatIndex(const std::vector<uint32_t>& index, in indexToFlatIndex()68 bool IndexedShapeWrapper::broadcastedIndexToFlatIndex(const std::vector<uint32_t>& index, in broadcastedIndexToFlatIndex()84 bool IndexedShapeWrapper::isValid(const std::vector<uint32_t>& index) const { in isValid()
152 "IndexedShapeWrapper.cpp",
39 IndexedShapeWrapper aShapeIndexed(aShape); in evalGeneric()40 IndexedShapeWrapper bShapeIndexed(bShape); in evalGeneric()41 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()65 IndexedShapeWrapper aShapeIndexed(aShape); in evalQuant8()66 IndexedShapeWrapper bShapeIndexed(bShape); in evalQuant8()67 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalQuant8()
37 IndexedShapeWrapper baseShapeIndexed(baseShape); in evalGeneric()38 IndexedShapeWrapper exponentShapeIndexed(exponentShape); in evalGeneric()39 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()
41 IndexedShapeWrapper aShapeIndexed(aShape); in compute()42 IndexedShapeWrapper bShapeIndexed(bShape); in compute()43 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
56 const IndexedShapeWrapper indexedOutput = IndexedShapeWrapper(outputShape); in evalGeneric()57 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()
46 IndexedShapeWrapper aShapeIndexed(aShape); in compute()47 IndexedShapeWrapper bShapeIndexed(bShape); in compute()48 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
212 IndexedShapeWrapper aShapeIndexed(aShape); in executeInt32()213 IndexedShapeWrapper bShapeIndexed(bShape); in executeInt32()214 IndexedShapeWrapper outputShapeIndexed(outputShape); in executeInt32()
30 class IndexedShapeWrapper {32 IndexedShapeWrapper(const Shape& wrapped_shape);