Searched refs:specify_workspace_limit (Results 1 – 1 of 1) sorted by relevance
2414 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo() argument2426 size_t mem_limit = specify_workspace_limit ? memory_limit_bytes : 0ULL; in GetCudnnConvolutionForwardAlgo()2439 specify_workspace_limit ? CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT in GetCudnnConvolutionForwardAlgo()2455 bool specify_workspace_limit, in GetCudnnConvolutionBackwardDataAlgo() argument2467 size_t mem_limit = specify_workspace_limit ? memory_limit_bytes : 0ULL; in GetCudnnConvolutionBackwardDataAlgo()2481 specify_workspace_limit in GetCudnnConvolutionBackwardDataAlgo()2498 bool specify_workspace_limit, in GetCudnnConvolutionBackwardFilterAlgo() argument2509 size_t mem_limit = specify_workspace_limit ? memory_limit_bytes : 0ULL; in GetCudnnConvolutionBackwardFilterAlgo()2523 specify_workspace_limit in GetCudnnConvolutionBackwardFilterAlgo()2708 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionForwardAlgorithm() local[all …]