Searched refs:a_reshape_dims (Results 1 – 1 of 1) sorted by relevance
1720 SmallVector<int64_t, 4> a_reshape_dims; in convertSpaceToDepthOp() local1721 a_reshape_dims.push_back(input_shape[0]); in convertSpaceToDepthOp()1722 a_reshape_dims.push_back(input_shape[1] / block_size[0]); in convertSpaceToDepthOp()1723 a_reshape_dims.push_back(block_size[0]); in convertSpaceToDepthOp()1724 a_reshape_dims.push_back(input_shape[2] / block_size[1]); in convertSpaceToDepthOp()1725 a_reshape_dims.push_back(block_size[1]); in convertSpaceToDepthOp()1726 a_reshape_dims.push_back(input_shape[3]); in convertSpaceToDepthOp()1729 ArrayRef<int64_t>(a_reshape_dims), output_type.getElementType()); in convertSpaceToDepthOp()1732 rewriter.getI64ArrayAttr(a_reshape_dims)); in convertSpaceToDepthOp()1805 SmallVector<int64_t, 4> a_reshape_dims; in convertDepthToSpaceOp() local[all …]