| /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)
253 ksize = test_util.NHWCToNCHW(ksize)
254 strides = test_util.NHWCToNCHW(strides)
255 t = test_util.NHWCToNCHW(t)
256 output_sizes = test_util.NHWCToNCHW(output_sizes)
|
| D | conv_ops_test.py | 208 t1 = test_util.NHWCToNCHW(t1)
209 strides = test_util.NHWCToNCHW(strides)
210 dilations = test_util.NHWCToNCHW(dilations)
212 padding = test_util.NHWCToNCHW(padding)
246 t1 = test_util.NHWCToNCHW(t1)
247 strides = test_util.NHWCToNCHW(strides)
274 t1 = test_util.NHWCToNCHW(t1)
721 input_sizes = test_util.NHWCToNCHW(input_sizes)
730 t2 = test_util.NHWCToNCHW(t2)
731 strides = test_util.NHWCToNCHW(strides)
[all …]
|
| D | conv_ops_3d_test.py | 87 t1 = test_util.NHWCToNCHW(t1)
88 strides = test_util.NHWCToNCHW(strides)
144 t1 = test_util.NHWCToNCHW(t1)
423 input_tensor = test_util.NHWCToNCHW(orig_input_tensor)
424 new_strides = test_util.NHWCToNCHW(strides)
705 input_sizes = test_util.NHWCToNCHW(input_sizes)
711 full_strides = test_util.NHWCToNCHW(full_strides)
712 full_dilations = test_util.NHWCToNCHW(full_dilations)
|
| D | pooling_ops_test.py | 151 t = test_util.NHWCToNCHW(t)
152 ksize = test_util.NHWCToNCHW(ksize)
153 strides = test_util.NHWCToNCHW(strides)
995 t = test_util.NHWCToNCHW(input_tensor)
1071 t = test_util.NHWCToNCHW(input_tensor)
1084 t = test_util.NHWCToNCHW(t)
|
| D | depthtospace_op_test.py | 50 input_nchw = test_util.NHWCToNCHW(input_nhwc)
|
| D | spacetodepth_op_test.py | 50 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 | 49 template struct NHWCToNCHW<Eigen::GpuDevice, float, 4>; variable
56 template struct NHWCToNCHW<Eigen::GpuDevice, float, 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 | cudnn_pooling_gpu.cc | 54 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
175 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
180 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
184 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
|
| D | pooling_ops_common.cc | 186 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
358 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
366 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
370 functor::NHWCToNCHW<GPUDevice, T, 4>()( in Compute()
|
| D | fused_batch_norm_op.cc | 285 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
425 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
437 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_ops_3d.cc | 338 functor::NHWCToNCHW<GPUDevice, T, 5>()( in launch()
533 void NHWCToNCHW<GPUDevice, T, 5>::operator()( \
|
| D | conv_2d.h | 270 struct NHWCToNCHW {
|
| D | conv_grad_filter_ops.cc | 773 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
794 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_grad_ops_3d.cc | 1309 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
1702 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
1720 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
|
| D | conv_2d_gpu.h | 977 struct NHWCToNCHW<GPUDevice, T, NDIMS> {
|
| D | conv_ops_fused_impl.h | 655 functor::NHWCToNCHW<GPUDevice, T, 4>()(
|
| D | conv_ops.cc | 749 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_grad_input_ops.cc | 893 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test_base.py | 248 t1 = test_util.NHWCToNCHW(t1)
249 strides = test_util.NHWCToNCHW(strides)
293 t1 = test_util.NHWCToNCHW(t1)
294 strides = test_util.NHWCToNCHW(strides)
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 406 functor::NHWCToNCHW<GPUDevice, T, NDIMS>()( in TransformNHWCToNCHW()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 847 bool NHWCToNCHW) { in AddNodeTranspose() argument
864 if (NHWCToNCHW) { in AddNodeTranspose()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 267 def NHWCToNCHW(input_tensor): function
|