Searched refs:dimensionIdx (Results 1 – 2 of 2) sorted by relevance
73 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension() argument74 CHECK(0 <= dimensionIdx && dimensionIdx < shape.dimensions.size()); in getSizeOfDimension()75 return shape.dimensions[dimensionIdx]; in getSizeOfDimension()
178 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);