Home
last modified time | relevance | path

Searched refs:GetConvolveAlgorithms (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.cc270 bool StreamExecutor::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::StreamExecutor
282 return dnn_support->GetConvolveAlgorithms( in GetConvolveAlgorithms()
H A Dstream_executor_pimpl.h368 bool GetConvolveAlgorithms(dnn::ConvolutionKind kind,
H A Ddnn.cc110 bool DnnSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::dnn::DnnSupport
H A Ddnn.h1385 virtual bool GetConvolveAlgorithms(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h216 bool GetConvolveAlgorithms(
H A Dcuda_dnn.cc4758 got_algos = GetConvolveAlgorithms(cuda_compute_capability, &algorithms); in GetConvolveRunners()
5220 if (!GetConvolveAlgorithms(cuda_compute_capability, &algorithms)) { in GetFusedConvolveRunners()
5274 bool CudnnSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::gpu::CudnnSupport
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h236 bool GetConvolveAlgorithms(
H A Drocm_dnn.cc3131 bool MIOpenSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::gpu::MIOpenSupport