| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 32 def NHWCToNCHW(input_tensor): function
97 t = NHWCToNCHW(t)
98 ksize = NHWCToNCHW(ksize)
99 strides = NHWCToNCHW(strides)
390 xla_inputs = NHWCToNCHW(inputs)
391 xla_outputs = NHWCToNCHW(outputs)
392 xla_output_gradients = NHWCToNCHW(output_gradients)
393 xla_output_grad_gradients = NHWCToNCHW(output_grad_gradients)
394 xla_ksize = NHWCToNCHW(ksize)
395 xla_strides = NHWCToNCHW(strides)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_3d_test.py | 73 t = test_util.NHWCToNCHW(t)
74 window = test_util.NHWCToNCHW(window)
75 strides = test_util.NHWCToNCHW(strides)
273 ksize = test_util.NHWCToNCHW(ksize)
274 strides = test_util.NHWCToNCHW(strides)
275 t = test_util.NHWCToNCHW(t)
276 output_sizes = test_util.NHWCToNCHW(output_sizes)
|
| D | conv_ops_test.py | 212 t1 = test_util.NHWCToNCHW(t1)
213 strides = test_util.NHWCToNCHW(strides)
214 dilations = test_util.NHWCToNCHW(dilations)
216 padding = test_util.NHWCToNCHW(padding)
251 t1 = test_util.NHWCToNCHW(t1)
252 strides = test_util.NHWCToNCHW(strides)
279 t1 = test_util.NHWCToNCHW(t1)
809 t1 = test_util.NHWCToNCHW(t1)
810 strides = test_util.NHWCToNCHW(strides)
811 dilations = test_util.NHWCToNCHW(dilations)
[all …]
|
| D | conv_ops_3d_test.py | 91 t1 = test_util.NHWCToNCHW(t1)
92 strides = test_util.NHWCToNCHW(strides)
144 t1 = test_util.NHWCToNCHW(t1)
519 input_tensor = test_util.NHWCToNCHW(orig_input_tensor)
520 new_strides = test_util.NHWCToNCHW(strides)
801 input_sizes = test_util.NHWCToNCHW(input_sizes)
807 full_strides = test_util.NHWCToNCHW(full_strides)
808 full_dilations = test_util.NHWCToNCHW(full_dilations)
|
| D | pooling_ops_test.py | 165 t = test_util.NHWCToNCHW(t)
166 ksize = test_util.NHWCToNCHW(ksize)
167 strides = test_util.NHWCToNCHW(strides)
169 padding = test_util.NHWCToNCHW(padding)
1160 padding = test_util.NHWCToNCHW(padding)
1161 t = test_util.NHWCToNCHW(input_tensor)
1237 t = test_util.NHWCToNCHW(input_tensor)
1250 t = test_util.NHWCToNCHW(t)
|
| D | depthtospace_op_test.py | 49 input_nchw = test_util.NHWCToNCHW(input_nhwc)
64 input_nchw = test_util.NHWCToNCHW(input_nhwc)
|
| D | spacetodepth_op_test.py | 51 input_nchw = test_util.NHWCToNCHW(input_nhwc)
|
| /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_float.cu.cc | 45 template struct NHWCToNCHW<Eigen::GpuDevice, float, 4>; variable
52 template struct NHWCToNCHW<Eigen::GpuDevice, float, 5>; variable
|
| 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 | cudnn_pooling_gpu.cc | 54 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
190 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
195 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
199 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
|
| D | pooling_ops_common.cc | 250 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
518 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
527 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
531 functor::NHWCToNCHW<GPUDevice, T, 4>()( in Compute()
|
| D | lrn_op.cc | 245 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<GPUDevice>(), in launch()
539 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<GPUDevice>(), in launch()
549 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<GPUDevice>(), in launch()
559 functor::NHWCToNCHW<GPUDevice, T, 4>()( in launch()
|
| D | conv_ops_3d.cc | 366 functor::NHWCToNCHW<GPUDevice, T, 5>()( in launch()
668 void NHWCToNCHW<GPUDevice, T, 5>::operator()( \
|
| D | conv_2d.h | 358 struct NHWCToNCHW {
|
| D | conv_grad_filter_ops.cc | 908 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
931 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| D | fused_batch_norm_op.cc | 867 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
1044 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
1056 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_grad_ops_3d.cc | 1362 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
1853 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
1872 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
|
| D | conv_grad_input_ops.cc | 323 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_ops_fused_impl.h | 549 functor::NHWCToNCHW<GPUDevice, T, 4>()(
|
| D | conv_2d_gpu.h | 1100 struct NHWCToNCHW<GPUDevice, T, NDIMS> {
|
| D | conv_ops.cc | 846 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 868 bool NHWCToNCHW) { in AddNodeTranspose() argument
885 if (NHWCToNCHW) { in AddNodeTranspose()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 335 def NHWCToNCHW(input_tensor): function
|