Searched refs:SwapDimension0And2InTensor3 (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu_float.cu.cc | 38 template struct SwapDimension0And2InTensor3<Eigen::GpuDevice, float4>; variable
39 template struct SwapDimension0And2InTensor3<Eigen::GpuDevice, float2, variable
|
| D | conv_2d_gpu_uint8.cu.cc | 33 template struct SwapDimension0And2InTensor3<Eigen::GpuDevice, uint8>; variable
|
| D | conv_2d_gpu_uint64.cu.cc | 33 template struct SwapDimension0And2InTensor3<Eigen::GpuDevice, uint64>; variable
|
| D | conv_2d_gpu_uint16.cu.cc | 33 template struct SwapDimension0And2InTensor3<Eigen::GpuDevice, uint16>; variable
|
| D | conv_2d_gpu_uint32.cu.cc | 33 template struct SwapDimension0And2InTensor3<Eigen::GpuDevice, uint32>; variable
|
| D | conv_2d_gpu_double.cu.cc | 30 template struct SwapDimension0And2InTensor3<Eigen::GpuDevice, double2, variable
|
| D | transpose_functor_gpu.cu.cc | 129 tensorflow::functor::SwapDimension0And2InTensor3<GPUDevice, T, in run()
|
| D | conv_2d.h | 388 struct SwapDimension0And2InTensor3 {
|
| D | conv_2d_gpu.h | 1051 struct SwapDimension0And2InTensor3<GPUDevice, T, conjugate> {
|