Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.h142 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size); in AllocateScratchOrFallback() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_algorithm_picker.cc505 auto scratch_or = scratch_allocator.AllocateBytes(*workspace_size); in AutotuneOneConvRunner() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc3085 const auto scratch_or = AllocateCudnnConvolutionForwardWorkspace( in GetCudnnConvolutionForwardAlgorithm() local
3147 const auto scratch_or = AllocateCudnnConvolutionBackwardDataWorkspace( in GetCudnnConvolutionBackwardDataAlgorithm() local
3208 port::StatusOr<DeviceMemory<uint8>> scratch_or = in GetCudnnConvolutionBackwardFilterAlgorithm() local
5896 const auto scratch_or = in DoPrepareForCtcLoss() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2488 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size_in_bytes); in DoPrepareForCtcLoss() local