Searched refs:IndexedShapeWrapper (Results 1 – 9 of 9) 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()
124 "IndexedShapeWrapper.cpp",
33 IndexedShapeWrapper aShapeIndexed(aShape); in evalGeneric()34 IndexedShapeWrapper bShapeIndexed(bShape); in evalGeneric()35 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()59 IndexedShapeWrapper aShapeIndexed(aShape); in evalQuant8()60 IndexedShapeWrapper bShapeIndexed(bShape); in evalQuant8()61 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalQuant8()
34 IndexedShapeWrapper baseShapeIndexed(baseShape); in evalGeneric()35 IndexedShapeWrapper exponentShapeIndexed(exponentShape); in evalGeneric()36 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()
39 IndexedShapeWrapper aShapeIndexed(aShape); in compute()40 IndexedShapeWrapper bShapeIndexed(bShape); in compute()41 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
50 const IndexedShapeWrapper indexedOutput = IndexedShapeWrapper(outputShape); in evalGeneric()51 const IndexedShapeWrapper indexedInput = IndexedShapeWrapper(inputShape); in evalGeneric()
43 IndexedShapeWrapper aShapeIndexed(aShape); in eval()44 IndexedShapeWrapper bShapeIndexed(bShape); in eval()45 IndexedShapeWrapper outputShapeIndexed(outputShape); in eval()
41 IndexedShapeWrapper aShapeIndexed(aShape); in compute()42 IndexedShapeWrapper bShapeIndexed(bShape); in compute()43 IndexedShapeWrapper outputShapeIndexed(outputShape); in compute()
30 class IndexedShapeWrapper {32 IndexedShapeWrapper(const Shape& wrapped_shape);