Searched defs:NHWCToNCHW (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu_double.cu.cc | 36 template struct NHWCToNCHW<Eigen::GpuDevice, double, 4>; variable
43 template struct NHWCToNCHW<Eigen::GpuDevice, double, 5>; variable
|
| D | conv_2d_gpu_half.cu.cc | 43 template struct NHWCToNCHW<Eigen::GpuDevice, Eigen::half, 4>; variable
50 template struct NHWCToNCHW<Eigen::GpuDevice, Eigen::half, 5>; variable
|
| D | conv_2d_gpu_float.cu.cc | 49 template struct NHWCToNCHW<Eigen::GpuDevice, float, 4>; variable
56 template struct NHWCToNCHW<Eigen::GpuDevice, float, 5>; variable
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 32 def NHWCToNCHW(input_tensor): function
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 267 def NHWCToNCHW(input_tensor): function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 847 bool NHWCToNCHW) { in AddNodeTranspose()
|