Searched defs:scratch_or (Results 1 – 4 of 4) sorted by relevance
142 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size); in AllocateScratchOrFallback() local
505 auto scratch_or = scratch_allocator.AllocateBytes(*workspace_size); in AutotuneOneConvRunner() local
3085 const auto scratch_or = AllocateCudnnConvolutionForwardWorkspace( in GetCudnnConvolutionForwardAlgorithm() local3147 const auto scratch_or = AllocateCudnnConvolutionBackwardDataWorkspace( in GetCudnnConvolutionBackwardDataAlgorithm() local3208 port::StatusOr<DeviceMemory<uint8>> scratch_or = in GetCudnnConvolutionBackwardFilterAlgorithm() local5896 const auto scratch_or = in DoPrepareForCtcLoss() local
2488 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size_in_bytes); in DoPrepareForCtcLoss() local