Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc264 bool StreamExecutor::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::StreamExecutor
273 return dnn_support->GetConvolveAlgorithms(with_winograd_nonfused, cc_major, in GetConvolveAlgorithms()
Dstream_executor_pimpl.h365 bool GetConvolveAlgorithms(bool with_winograd_nonfused,
Ddnn.cc44 bool DnnSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::dnn::DnnSupport
Ddnn.h1326 virtual bool GetConvolveAlgorithms(
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc116 succ = stream_exec->GetConvolveAlgorithms(true, &algorithms); in GetAlgorithms()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc518 stream->parent()->GetConvolveAlgorithms( in launch()
Dconv_ops_fused_impl.h377 if (!stream->parent()->GetConvolveAlgorithms(
Dconv_ops.cc998 stream->parent()->GetConvolveAlgorithms( in operator ()()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h203 bool GetConvolveAlgorithms(
Dcuda_dnn.cc3294 bool CudnnSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h230 bool GetConvolveAlgorithms(
Drocm_dnn.cc3082 bool MIOpenSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::gpu::MIOpenSupport