Searched defs:specify_workspace_limit (Results 1 – 1 of 1) sorted by relevance
2414 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo()2455 bool specify_workspace_limit, in GetCudnnConvolutionBackwardDataAlgo()2498 bool specify_workspace_limit, in GetCudnnConvolutionBackwardFilterAlgo()2708 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionForwardAlgorithm() local2770 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardDataAlgorithm() local2831 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardFilterAlgorithm() local