Searched defs:NHWCToNCHW (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu_half.cu.cc | 37 template struct NHWCToNCHW<Eigen::GpuDevice, Eigen::half, 4>; variable
44 template struct NHWCToNCHW<Eigen::GpuDevice, Eigen::half, 5>; variable
|
| 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_float.cu.cc | 45 template struct NHWCToNCHW<Eigen::GpuDevice, float, 4>; variable
52 template struct NHWCToNCHW<Eigen::GpuDevice, float, 5>; variable
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | SpaceToDepthTestImpl.cpp | 36 const armnn::PermutationVector NHWCToNCHW = {0, 2, 3, 1}; in SpaceToDepthTestImpl() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 28 def NHWCToNCHW(input_tensor): function
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 369 def NHWCToNCHW(input_tensor): function
|