HomeSort by relevance Sort by last modified time
    Searched defs:LaunchDepthwiseConvOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
depthwise_conv_op.cc 153 struct LaunchDepthwiseConvOp<CPUDevice, T> {
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,
depthwise_conv_op.h 63 struct LaunchDepthwiseConvOp {
85 struct LaunchDepthwiseConvOp<Eigen::GpuDevice, T> {

Completed in 65 milliseconds