Home
last modified time | relevance | path

Searched defs:specify_workspace_limit (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2706 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo()
2747 bool specify_workspace_limit, in GetCudnnConvolutionBackwardDataAlgo()
2790 bool specify_workspace_limit, in GetCudnnConvolutionBackwardFilterAlgo()
3000 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionForwardAlgorithm() local
3062 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardDataAlgorithm() local
3123 bool specify_workspace_limit = scratch_allocator != nullptr; in GetCudnnConvolutionBackwardFilterAlgorithm() local
3843 bool specify_workspace_limit = scratch_allocator != nullptr; in GetFirstWorkingExecutionPlan() local