Home
last modified time | relevance | path

Searched refs:GetConvolveBackwardDataAlgorithms (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc345 bool StreamExecutor::GetConvolveBackwardDataAlgorithms( in GetConvolveBackwardDataAlgorithms() function in stream_executor::StreamExecutor
354 return dnn_support->GetConvolveBackwardDataAlgorithms( in GetConvolveBackwardDataAlgorithms()
Ddnn.cc39 bool DnnSupport::GetConvolveBackwardDataAlgorithms( in GetConvolveBackwardDataAlgorithms() function in stream_executor::dnn::DnnSupport
Dstream_executor_pimpl.h392 bool GetConvolveBackwardDataAlgorithms(
Ddnn.h1364 virtual bool GetConvolveBackwardDataAlgorithms(
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_algorithm_picker.cc54 succ = stream_exec->GetConvolveBackwardDataAlgorithms(true, &algorithms); in GetAlgorithms()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h209 bool GetConvolveBackwardDataAlgorithms(
Dcuda_dnn.cc3103 bool CudnnSupport::GetConvolveBackwardDataAlgorithms( in GetConvolveBackwardDataAlgorithms() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h200 bool GetConvolveBackwardDataAlgorithms(
Drocm_dnn.cc2968 bool MIOpenSupport::GetConvolveBackwardDataAlgorithms( in GetConvolveBackwardDataAlgorithms() function in stream_executor::gpu::MIOpenSupport
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc955 CHECK(stream->parent()->GetConvolveBackwardDataAlgorithms( in operator ()()
Dconv_grad_ops_3d.cc1361 CHECK(stream->parent()->GetConvolveBackwardDataAlgorithms( in Compute()