Home
last modified time | relevance | path

Searched refs:launcher_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc405 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, in Compute()
445 LaunchConv2DOp<Device, T> launcher_; member in tensorflow::DepthwiseConv2dNativeOp
Ddepthwise_conv_grad_op.cc634 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, in Compute()
659 LaunchConv2DBackpropInputOp<Device, T> launcher_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1117 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
1142 LaunchConv2DBackpropFilterOp<Device, T> launcher_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp
Dconv_ops.cc517 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter, in Compute()
528 LaunchConv2DOp<Device, T> launcher_; member in tensorflow::Conv2DOp
Dconv_grad_filter_ops.cc524 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
536 LaunchConv2DBackpropFilterOp<Device, T> launcher_; member in tensorflow::Conv2DSlowBackpropFilterOp
Dconv_grad_input_ops.cc661 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, filter, in Compute()
673 LaunchConv2DBackpropInputOp<Device, T> launcher_; member in tensorflow::Conv2DSlowBackpropInputOp
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc240 launcher_.launch(context, cudnn_use_autotune_, conv_input, conv_input_scale, in Compute()
253 LaunchFusedConv2DBiasActivationOp<Device, T, BiasType, ScaleType> launcher_; member in tensorflow::FusedConv2DBiasActivationOp