Searched refs:spaceToDepthPrepare (Results 1 – 3 of 3) sorted by relevance
311 bool spaceToDepthPrepare(const Shape& input, int32_t blockSize, Shape* output);
441 bool spaceToDepthPrepare(const Shape& input, int32_t blockSize, Shape* output) { in spaceToDepthPrepare() function
952 if (!spaceToDepthPrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()