Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3012 const auto scratch_or = AllocateCudnnConvolutionForwardWorkspace( in GetCudnnConvolutionForwardAlgorithm() local
3074 const auto scratch_or = AllocateCudnnConvolutionBackwardDataWorkspace( in GetCudnnConvolutionBackwardDataAlgorithm() local
3135 port::StatusOr<DeviceMemory<uint8>> scratch_or = in GetCudnnConvolutionBackwardFilterAlgorithm() local
4109 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size); in DoConvolveWithExecutionPlan() local
5071 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size); in DoFusedConvolveWithExecutionPlan() local
5206 const auto scratch_or = in DoPrepareForCtcLoss() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2485 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size_in_bytes); in DoPrepareForCtcLoss() local