Searched refs:scratch_or (Results 1 – 1 of 1) sorted by relevance
2364 const auto scratch_or = AllocateCudnnConvolutionForwardWorkspace( in GetCudnnConvolutionForwardAlgorithm() local2368 if (scratch_or.ok()) { in GetCudnnConvolutionForwardAlgorithm()2369 *scratch = scratch_or.ValueOrDie(); in GetCudnnConvolutionForwardAlgorithm()2413 const auto scratch_or = AllocateCudnnConvolutionBackwardDataWorkspace( in GetCudnnConvolutionBackwardDataAlgorithm() local2417 if (scratch_or.ok()) { in GetCudnnConvolutionBackwardDataAlgorithm()2418 *scratch = scratch_or.ValueOrDie(); in GetCudnnConvolutionBackwardDataAlgorithm()2462 auto scratch_or = AllocateCudnnConvolutionBackwardFilterWorkspace( in GetCudnnConvolutionBackwardFilterAlgorithm() local2466 if (scratch_or.ok()) { in GetCudnnConvolutionBackwardFilterAlgorithm()2467 *scratch = scratch_or.ValueOrDie(); in GetCudnnConvolutionBackwardFilterAlgorithm()