Searched defs:LaunchDepthwiseConvOp (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | depthwise_conv_op.h | 63 struct LaunchDepthwiseConvOp { struct in namespace:tensorflow 85 struct LaunchDepthwiseConvOp<Eigen::GpuDevice, T> { struct in namespace:tensorflow
|
H A D | depthwise_conv_op.cc | 153 struct LaunchDepthwiseConvOp<CPUDevice, T> { struct in namespace:tensorflow 249 extern template struct LaunchDepthwiseConvOp<GPUDevice, Eigen::half>; 250 extern template struct LaunchDepthwiseConvOp<GPUDevice, float>; 251 extern template struct LaunchDepthwiseConvOp<GPUDevice, double>; 402 LaunchDepthwiseConvOp<Device, T>()(context, args, input_ptr, filter_ptr,
|
Completed in 111 milliseconds