cudnnTensorDescriptor_t data_desc_;
cudnnDropoutDescriptor_t dropout_desc_;
- at::IntList cudnn_input_dims_;
+ vector<int64_t> cudnn_input_dims_;
float ratio_;
bool is_test_;
cudnnTensorDescriptor_t data_desc_;
cudnnDropoutDescriptor_t dropout_desc_;
- at::IntList cudnn_input_dims_;
+ vector<int64_t> cudnn_input_dims_;
Blob* scratch_blob_;
if (X.sizes() != cudnn_input_dims_) {
CAFFE_ENFORCE(scratch_blob_);
Tensor* states = BlobGetMutableTensor(scratch_blob_, CUDA);
- cudnn_input_dims_ = X.sizes();
+ cudnn_input_dims_ = X.sizes().vec();
CUDNN_ENFORCE(cudnnSetTensor4dDescriptor(
data_desc_,
GetCudnnTensorFormat(StorageOrder::NCHW),
}
if (dY.sizes() != cudnn_input_dims_) {
- cudnn_input_dims_ = dY.sizes();
+ cudnn_input_dims_ = dY.sizes().vec();
CUDNN_ENFORCE(cudnnSetTensor4dDescriptor(
data_desc_,
GetCudnnTensorFormat(StorageOrder::NCHW),