Searched refs:LaunchDepthwiseConvOp (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 154 struct LaunchDepthwiseConvOp<CPUDevice, T> { struct
257 extern template struct LaunchDepthwiseConvOp<GPUDevice, Eigen::half>;
258 extern template struct LaunchDepthwiseConvOp<GPUDevice, float>;
259 extern template struct LaunchDepthwiseConvOp<GPUDevice, double>;
430 LaunchDepthwiseConvOp<Device, T>()(context, args, input_ptr, filter_ptr, in Compute()
|
| D | depthwise_conv_op_gpu_float.cu.cc | 25 template struct LaunchDepthwiseConvOp<GpuDevice, float>; variable
|
| D | depthwise_conv_op_gpu_double.cu.cc | 25 template struct LaunchDepthwiseConvOp<GpuDevice, double>; variable
|
| D | depthwise_conv_op_gpu_half.cu.cc | 25 template struct LaunchDepthwiseConvOp<GpuDevice, Eigen::half>; variable
|
| D | depthwise_conv_op.h | 63 struct LaunchDepthwiseConvOp { struct
85 struct LaunchDepthwiseConvOp<Eigen::GpuDevice, T> {
|
| D | depthwise_conv_op_gpu.h | 798 void LaunchDepthwiseConvOp<GpuDevice, T>::operator()(OpKernelContext* ctx,
|