Home
last modified time | relevance | path

Searched refs:cudnn_launch_status (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc456 bool cudnn_launch_status = in launch() local
463 if (cudnn_launch_status) { in launch()
485 bool cudnn_launch_status = in launch() local
493 if (!cudnn_launch_status) { in launch()
Dconv_grad_ops_3d.cc1373 bool cudnn_launch_status = in Compute() local
1380 if (cudnn_launch_status) { in Compute()
1409 bool cudnn_launch_status = in Compute() local
1417 if (!cudnn_launch_status) { in Compute()
1779 bool cudnn_launch_status = in Compute() local
1787 if (cudnn_launch_status) { in Compute()
1816 bool cudnn_launch_status = in Compute() local
1824 if (!cudnn_launch_status) { in Compute()
Dfused_batch_norm_op.cc358 bool cudnn_launch_status = in operator ()() local
369 if (!cudnn_launch_status) { in operator ()()
484 bool cudnn_launch_status = in operator ()() local
492 if (!cudnn_launch_status) { in operator ()()
Dconv_ops_fused_impl.h529 bool cudnn_launch_status =
533 if (cudnn_launch_status && profile_result.is_valid()) {
807 bool cudnn_launch_status = launch(algorithm_config, &scratch_allocator,
810 context, cudnn_launch_status,
Dconv_ops.cc866 bool cudnn_launch_status = in operator ()() local
873 if (cudnn_launch_status) { in operator ()()
895 bool cudnn_launch_status = in operator ()() local
903 if (!cudnn_launch_status) { in operator ()()
Dconv_grad_filter_ops.cc853 bool cudnn_launch_status = in operator ()() local
861 if (cudnn_launch_status) { in operator ()()
884 bool cudnn_launch_status = in operator ()() local
892 if (!cudnn_launch_status) { in operator ()()
Dconv_grad_input_ops.cc965 bool cudnn_launch_status = in operator ()() local
972 if (cudnn_launch_status) { in operator ()()
994 bool cudnn_launch_status = in operator ()() local
1002 if (!cudnn_launch_status) { in operator ()()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc720 bool cudnn_launch_status = in launch() local
729 if (cudnn_launch_status) { in launch()
754 bool cudnn_launch_status = in launch() local
764 if (!cudnn_launch_status) { in launch()