| /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_float.cu.cc | 46 template struct NCHWToNHWC<Eigen::GpuDevice, float, 4>; variable
53 template struct NCHWToNHWC<Eigen::GpuDevice, float, 5>; variable
|
| D | conv_2d_gpu_half.cu.cc | 38 template struct NCHWToNHWC<Eigen::GpuDevice, Eigen::half, 4>; variable
45 template struct NCHWToNHWC<Eigen::GpuDevice, Eigen::half, 5>; variable
|
| D | cudnn_pooling_gpu.cc | 125 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
270 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
|
| D | pooling_ops_common.cc | 414 functor::NCHWToNHWC<GPUDevice, RT, 4>()( in Compute()
750 functor::NCHWToNHWC<GPUDevice, T, 4>()( in Compute()
|
| D | conv_ops_3d.cc | 637 functor::NCHWToNHWC<GPUDevice, T, 5>()( in launch()
672 void NCHWToNHWC<GPUDevice, T, 5>::operator()( \
|
| D | lrn_op.cc | 289 functor::NCHWToNHWC<GPUDevice, T, 4>()( in launch()
619 functor::NCHWToNHWC<GPUDevice, T, 4>()( in launch()
|
| D | conv_2d.h | 368 struct NCHWToNHWC {
|
| D | conv_grad_input_ops.cc | 543 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_ops_fused_impl.h | 707 functor::NCHWToNHWC<GPUDevice, T, 4>()(
|
| D | fused_batch_norm_op.cc | 975 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
1141 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_2d_gpu.h | 1118 struct NCHWToNHWC<GPUDevice, T, NDIMS> {
|
| D | conv_ops.cc | 1126 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_grad_ops_3d.cc | 1557 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 47 def NCHWToNHWC(input_tensor): function
106 t = NCHWToNHWC(t)
405 actual_input_gradients = NCHWToNHWC(actual_input_gradients)
416 actual_grad_gradients = NCHWToNHWC(actual_grad_gradients)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_3d_test.py | 96 conv = test_util.NCHWToNHWC(conv)
165 expected = test_util.NCHWToNHWC(expected)
166 computed = test_util.NCHWToNHWC(computed)
534 conv = test_util.NCHWToNHWC(conv)
824 actual = test_util.NCHWToNHWC(actual)
825 expected = test_util.NCHWToNHWC(expected)
|
| D | depthtospace_op_test.py | 52 output_nhwc = test_util.NCHWToNHWC(output_nchw)
67 output_nhwc = test_util.NCHWToNHWC(output_nchw)
|
| D | conv_ops_test.py | 226 conv = test_util.NCHWToNHWC(conv)
256 conv = test_util.NCHWToNHWC(conv)
300 expected = test_util.NCHWToNHWC(expected)
301 computed = test_util.NCHWToNHWC(computed)
941 conv = test_util.NCHWToNHWC(conv)
975 conv = test_util.NCHWToNHWC(conv)
1319 conv_forward = test_util.NCHWToNHWC(conv_forward)
1320 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_2)
1365 conv_forward = test_util.NCHWToNHWC(conv_forward)
1366 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_2)
[all …]
|
| D | pooling_ops_3d_test.py | 83 t = test_util.NCHWToNHWC(t)
|
| D | spacetodepth_op_test.py | 54 output_nhwc = test_util.NCHWToNHWC(output_nchw)
|
| D | pooling_ops_test.py | 190 t = test_util.NCHWToNHWC(t)
1174 t = test_util.NCHWToNHWC(t)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 421 def NCHWToNHWC(input_tensor): function
|