D | cuda_dnn.cc | 51 #pragma clang diagnostic push 55 #pragma clang diagnostic warning "-Wmismatched-tags" 552 cudnnTensorFormat_t format; in CudnnFilterDescriptor() local 555 format = CUDNN_TENSOR_NCHW; in CudnnFilterDescriptor() 558 format = CUDNN_TENSOR_NHWC; in CudnnFilterDescriptor() 561 format = CUDNN_TENSOR_NCHW_VECT_C; in CudnnFilterDescriptor() 576 CHECK_CUDNN_OK(cudnnSetFilterNdDescriptor(handle_.get(), elem_type, format, in CudnnFilterDescriptor() 1209 for (int region = 0; region < region_count_per_layer; region++) { in Create() local 1217 /*w=*/nullptr, /*linLayerID=*/region, in Create() 1224 /*w=*/nullptr, /*linLayerID=*/region, in Create() [all …]
|