Searched defs:specify_workspace_limit (Results 1 – 1 of 1) sorted by relevance
2766 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo()2807 bool specify_workspace_limit, in GetCudnnConvolutionBackwardDataAlgo()2850 bool specify_workspace_limit, in GetCudnnConvolutionBackwardFilterAlgo()3073 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionForwardAlgorithm() local3135 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardDataAlgorithm() local3196 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardFilterAlgorithm() local