Searched defs:specify_workspace_limit (Results 1 – 1 of 1) sorted by relevance
2159 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo()2177 bool specify_workspace_limit, in GetCudnnConvolutionBackwardDataAlgo()2196 bool specify_workspace_limit, in GetCudnnConvolutionBackwardFilterAlgo()2352 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionForwardAlgorithm() local2401 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardDataAlgorithm() local2450 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardFilterAlgorithm() local