Searched defs:scratch_or (Results 1 – 2 of 2) sorted by relevance
2720 const auto scratch_or = AllocateCudnnConvolutionForwardWorkspace( in GetCudnnConvolutionForwardAlgorithm() local2782 const auto scratch_or = AllocateCudnnConvolutionBackwardDataWorkspace( in GetCudnnConvolutionBackwardDataAlgorithm() local2843 auto scratch_or = AllocateCudnnConvolutionBackwardFilterWorkspace( in GetCudnnConvolutionBackwardFilterAlgorithm() local3945 const auto scratch_or = in DoPrepareForCtcLoss() local
2434 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size_in_bytes); in DoPrepareForCtcLoss() local