Searched defs:LaunchDepthwiseConvOp (Results 1 – 5 of 5) sorted by relevance
25 template struct LaunchDepthwiseConvOp<GpuDevice, float>; variable
25 template struct LaunchDepthwiseConvOp<GpuDevice, Eigen::half>; variable
25 template struct LaunchDepthwiseConvOp<GpuDevice, double>; variable
154 struct LaunchDepthwiseConvOp<CPUDevice, T> { struct155 typedef typename Eigen::internal::packet_traits<T>::type Packet;157 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args, in operator ()()
63 struct LaunchDepthwiseConvOp { struct