Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc323 bool StreamExecutor::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::StreamExecutor
332 return dnn_support->GetConvolveAlgorithms(with_winograd_nonfused, cc_major, in GetConvolveAlgorithms()
Ddnn.cc29 bool DnnSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::dnn::DnnSupport
Dstream_executor_pimpl.h385 bool GetConvolveAlgorithms(bool with_winograd_nonfused,
Ddnn.h1271 virtual bool GetConvolveAlgorithms(
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_algorithm_picker.cc58 succ = stream_exec->GetConvolveAlgorithms(true, &algorithms); in GetAlgorithms()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc441 stream->parent()->GetConvolveAlgorithms( in launch()
Dconv_ops_fused_impl.h515 if (!stream->parent()->GetConvolveAlgorithms(
Dconv_ops.cc853 stream->parent()->GetConvolveAlgorithms( in operator ()()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc696 CHECK(stream->parent()->GetConvolveAlgorithms( in launch()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h202 bool GetConvolveAlgorithms(
Dcuda_dnn.cc3042 bool CudnnSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h193 bool GetConvolveAlgorithms(
Drocm_dnn.cc2947 bool MIOpenSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms() function in stream_executor::gpu::MIOpenSupport