Searched refs:TransformDepth (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dpooling_ops_common.cc130 void TransformDepth<GPUDevice, T, Eigen::DenseIndex>::operator()( \
134 extern template struct TransformDepth<GPUDevice, T, Eigen::DenseIndex>;
H A Dconv_2d.h173 struct TransformDepth { struct in namespace:tensorflow::functor
H A Dconv_ops_gpu_3.cu.cc1018 template struct functor::TransformDepth<GPUDevice, float, int>; member in class:tensorflow::functor
1019 template struct functor::TransformDepth<GPUDevice, Eigen::half, int>; member in class:tensorflow::functor
H A Dconv_grad_filter_ops.cc820 // The first TransformDepth performs
823 // the second TransformDepth performs
1005 void TransformDepth<GPUDevice, T, int>::operator()( \
1009 extern template struct TransformDepth<GPUDevice, T, int>; \
H A Dconv_grad_input_ops.cc874 // The first TransformDepth performs
877 // the second TransformDepth performs
1078 void TransformDepth<GPUDevice, T, int>::operator()( \
1082 extern template struct TransformDepth<GPUDevice, T, int>; \

Completed in 201 milliseconds