Searched refs:SwapDimension1And2InTensor3 (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu_float.cu.cc | 34 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, float4>; variable
35 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, float2, variable
|
| D | conv_2d_gpu_uint8.cu.cc | 32 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, uint8>; variable
|
| D | conv_2d_gpu_uint64.cu.cc | 32 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, uint64>; variable
|
| D | conv_2d_gpu_uint16.cu.cc | 32 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, uint16>; variable
|
| D | conv_2d_gpu_uint32.cu.cc | 32 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, uint32>; variable
|
| D | conv_2d_gpu_double.cu.cc | 27 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, double2, variable
|
| D | conv_2d_gpu_half.cu.cc | 32 template struct SwapDimension1And2InTensor3<Eigen::GpuDevice, Eigen::half>; variable
|
| D | transpose_functor_gpu.cu.cc | 116 tensorflow::functor::SwapDimension1And2InTensor3<GPUDevice, T, in run()
124 tensorflow::functor::SwapDimension1And2InTensor3<GPUDevice, T, in run()
|
| D | conv_2d.h | 378 struct SwapDimension1And2InTensor3 {
|
| D | conv_2d_gpu.h | 1037 struct SwapDimension1And2InTensor3<GPUDevice, T, conjugate> {
|