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