Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:GetConvolveAlgorithms (Results 1 - 13 of 13) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_algorithm_picker.cc	58 succ = stream_exec->GetConvolveAlgorithms(true, &algorithms);
/external/tensorflow/tensorflow/stream_executor/
stream_executor_pimpl.cc	323 bool StreamExecutor::GetConvolveAlgorithms( 332 return dnn_support->GetConvolveAlgorithms(with_winograd_nonfused, cc_major,
dnn.cc	29 bool DnnSupport::GetConvolveAlgorithms(
stream_executor_pimpl.h	385 bool GetConvolveAlgorithms(bool with_winograd_nonfused,     [all...]
dnn.h	[all...]
/external/tensorflow/tensorflow/core/kernels/
conv_ops_3d.cc	441 stream->parent()->GetConvolveAlgorithms(
conv_ops_fused_impl.h	515 if (!stream->parent()->GetConvolveAlgorithms(     [all...]
conv_ops.cc	[all...]
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc	696 CHECK(stream->parent()->GetConvolveAlgorithms(     [all...]
/external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h	202 bool GetConvolveAlgorithms(     [all...]
cuda_dnn.cc	[all...]
/external/tensorflow/tensorflow/stream_executor/rocm/
rocm_dnn.h	193 bool GetConvolveAlgorithms(     [all...]
rocm_dnn.cc	[all...]

Completed in 648 milliseconds