/external/tensorflow/tensorflow/core/kernels/ |
D | depthwise_conv_op.cc | 405 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, in Compute() 445 LaunchConv2DOp<Device, T> launcher_; member in tensorflow::DepthwiseConv2dNativeOp
|
D | depthwise_conv_grad_op.cc | 634 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
|
D | conv_ops.cc | 517 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter, in Compute() 528 LaunchConv2DOp<Device, T> launcher_; member in tensorflow::Conv2DOp
|
D | conv_grad_filter_ops.cc | 524 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute() 536 LaunchConv2DBackpropFilterOp<Device, T> launcher_; member in tensorflow::Conv2DSlowBackpropFilterOp
|
D | conv_grad_input_ops.cc | 661 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/ |
D | fused_conv2d_bias_activation_op.cc | 240 launcher_.launch(context, cudnn_use_autotune_, conv_input, conv_input_scale, in Compute() 253 LaunchFusedConv2DBiasActivationOp<Device, T, BiasType, ScaleType> launcher_; member in tensorflow::FusedConv2DBiasActivationOp
|