Searched refs:TransformFilter (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_2d_gpu_double.cu.cc | 34 template struct TransformFilter<Eigen::GpuDevice, double, int, 4>; variable 41 template struct TransformFilter<Eigen::GpuDevice, double, int, 5>; variable
|
D | conv_2d_gpu_float.cu.cc | 43 template struct TransformFilter<Eigen::GpuDevice, float, int, 4>; variable 50 template struct TransformFilter<Eigen::GpuDevice, float, int, 5>; variable
|
D | conv_2d_gpu_half.cu.cc | 35 template struct TransformFilter<Eigen::GpuDevice, Eigen::half, int, 4>; variable 42 template struct TransformFilter<Eigen::GpuDevice, Eigen::half, int, 5>; variable
|
D | conv_2d_gpu_int.cu.cc | 33 template struct TransformFilter<Eigen::GpuDevice, int32, int, 4>; variable
|
D | conv_grad_input_ops.cc | 295 functor::TransformFilter<GPUDevice, T, int, 4>()( in operator ()() 556 void TransformFilter<GPUDevice, T, int, 4>::operator()( \ 560 extern template struct TransformFilter<GPUDevice, T, int, 4>; \
|
D | conv_ops_fused_impl.h | 611 functor::TransformFilter<GPUDevice, T, int, 4>()( 823 void TransformFilter<GPUDevice, T, int, 4>::operator()( \ 827 extern template struct TransformFilter<GPUDevice, T, int, 4>; \
|
D | conv_ops_3d.cc | 448 functor::TransformFilter<GPUDevice, T, int, 5>()( in launch() 651 void TransformFilter<GPUDevice, T, int, 5>::operator()( \
|
D | conv_ops.cc | 921 functor::TransformFilter<GPUDevice, T, int, 4>()( in operator ()() 1162 void TransformFilter<GPUDevice, T, int, 4>::operator()( \ 1166 extern template struct TransformFilter<GPUDevice, T, int, 4>; \
|
D | conv_2d.h | 207 struct TransformFilter {
|
D | conv_grad_filter_ops.cc | 1124 void TransformFilter<GPUDevice, T, int, 4>::operator()( \ 1128 extern template struct TransformFilter<GPUDevice, T, int, 4>; \
|
D | conv_grad_ops_3d.cc | 1071 void TransformFilter<GPUDevice, T, int, 5>::operator()( \ 1347 functor::TransformFilter<GPUDevice, T, int, 5>()( in Compute()
|
D | conv_2d_gpu.h | 482 struct TransformFilter<GPUDevice, T, int, NDIMS> {
|