Home
last modified time | relevance | path

Searched refs:GetConvolveRunners (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.cc318 TF_RETURN_IF_ERROR(stream->parent()->GetConvolveRunners( in AutotuneUnfusedConv()
362 TF_RETURN_IF_ERROR(stream->parent()->GetConvolveRunners( in AutotuneUnfusedConv()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream_executor_pimpl.cc293 port::Status StreamExecutor::GetConvolveRunners( in GetConvolveRunners() function in stream_executor::StreamExecutor
307 return dnn_support->GetConvolveRunners( in GetConvolveRunners()
Dstream_executor_pimpl.h372 port::Status GetConvolveRunners(
Ddnn.cc116 port::Status DnnSupport::GetConvolveRunners( in GetConvolveRunners() function in stream_executor::dnn::DnnSupport
Ddnn.h1389 virtual port::Status GetConvolveRunners(
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc162 TF_RETURN_IF_ERROR(stream_exec->GetConvolveRunners( in GetAlgorithms()
203 TF_RETURN_IF_ERROR(stream_exec->GetConvolveRunners( in GetMIOpenAlgorithms()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h220 port::Status GetConvolveRunners(
Dcuda_dnn.cc4710 port::Status CudnnSupport::GetConvolveRunners( in GetConvolveRunners() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h240 port::Status GetConvolveRunners(
Drocm_dnn.cc3146 port::Status MIOpenSupport::GetConvolveRunners( in GetConvolveRunners() function in stream_executor::gpu::MIOpenSupport