Lines Matching refs:shape
136 uint32_t getNumberOfElements(const Shape& shape) { in getNumberOfElements() argument
138 for (size_t i = 0; i < shape.dimensions.size(); i++) { in getNumberOfElements()
139 count *= shape.dimensions[i]; in getNumberOfElements()
144 uint32_t getNumberOfElements(const Shape& shape, size_t firstAxisInclusive, in getNumberOfElements() argument
148 nnAssert(lastAxisExclusive <= shape.dimensions.size()); in getNumberOfElements()
151 count *= shape.dimensions[i]; in getNumberOfElements()
156 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions() argument
157 return shape.dimensions.size(); in getNumberOfDimensions()
160 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension() argument
161 nnAssert(0 <= dimensionIdx && dimensionIdx < shape.dimensions.size()); in getSizeOfDimension()
162 return shape.dimensions[dimensionIdx]; in getSizeOfDimension()
165 uint32_t hasKnownRank(const Shape& shape) { in hasKnownRank() argument
166 return !shape.dimensions.empty(); in hasKnownRank()