Searched defs:NCHWToNHWC (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu_double.cu.cc | 37 template struct NCHWToNHWC<Eigen::GpuDevice, double, 4>; variable
44 template struct NCHWToNHWC<Eigen::GpuDevice, double, 5>; variable
|
| D | conv_2d_gpu_half.cu.cc | 44 template struct NCHWToNHWC<Eigen::GpuDevice, Eigen::half, 4>; variable
51 template struct NCHWToNHWC<Eigen::GpuDevice, Eigen::half, 5>; variable
|
| D | conv_2d_gpu_float.cu.cc | 50 template struct NCHWToNHWC<Eigen::GpuDevice, float, 4>; variable
57 template struct NCHWToNHWC<Eigen::GpuDevice, float, 5>; variable
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 47 def NCHWToNHWC(input_tensor): function
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 353 def NCHWToNHWC(input_tensor): function
|