| /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 | 50 template struct NCHWToNHWC<Eigen::GpuDevice, float, 4>; variable
57 template struct NCHWToNHWC<Eigen::GpuDevice, float, 5>; variable
|
| D | conv_2d_gpu_half.cu.cc | 44 template struct NCHWToNHWC<Eigen::GpuDevice, Eigen::half, 4>; variable
51 template struct NCHWToNHWC<Eigen::GpuDevice, Eigen::half, 5>; variable
|
| D | cudnn_pooling_gpu.cc | 110 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
239 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
|
| D | pooling_ops_common.cc | 276 functor::NCHWToNHWC<GPUDevice, RT, 4>()( in Compute()
428 functor::NCHWToNHWC<GPUDevice, T, 4>()( in Compute()
|
| D | conv_ops_3d.cc | 502 functor::NCHWToNHWC<GPUDevice, T, 5>()( in launch()
537 void NCHWToNHWC<GPUDevice, T, 5>::operator()( \
|
| D | fused_batch_norm_op.cc | 375 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
498 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_2d.h | 280 struct NCHWToNHWC {
|
| D | conv_2d_gpu.h | 995 struct NCHWToNHWC<GPUDevice, T, NDIMS> {
|
| D | conv_ops_fused_impl.h | 817 functor::NCHWToNHWC<GPUDevice, T, 4>()(
|
| D | conv_ops.cc | 911 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_grad_input_ops.cc | 1041 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_grad_ops_3d.cc | 1446 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_3d_test.py | 92 conv = test_util.NCHWToNHWC(conv)
165 expected = test_util.NCHWToNHWC(expected)
166 computed = test_util.NCHWToNHWC(computed)
438 conv = test_util.NCHWToNHWC(conv)
728 actual = test_util.NCHWToNHWC(actual)
729 expected = test_util.NCHWToNHWC(expected)
|
| D | conv_ops_test.py | 221 conv = test_util.NCHWToNHWC(conv)
251 conv = test_util.NCHWToNHWC(conv)
295 expected = test_util.NCHWToNHWC(expected)
296 computed = test_util.NCHWToNHWC(computed)
744 conv = test_util.NCHWToNHWC(conv)
778 conv = test_util.NCHWToNHWC(conv)
1106 conv_forward = test_util.NCHWToNHWC(conv_forward)
1107 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_2)
1152 conv_forward = test_util.NCHWToNHWC(conv_forward)
1153 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_2)
[all …]
|
| D | pooling_ops_3d_test.py | 83 t = test_util.NCHWToNHWC(t)
|
| D | depthtospace_op_test.py | 53 output_nhwc = test_util.NCHWToNHWC(output_nchw)
|
| D | spacetodepth_op_test.py | 53 output_nhwc = test_util.NCHWToNHWC(output_nchw)
|
| D | pooling_ops_test.py | 174 t = test_util.NCHWToNHWC(t)
1008 t = test_util.NCHWToNHWC(t)
|
| /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/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test_base.py | 265 output = test_util.NCHWToNHWC(output)
266 ref_output = test_util.NCHWToNHWC(ref_output)
305 output = test_util.NCHWToNHWC(output)
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 773 functor::NCHWToNHWC<GPUDevice, T, 4>()( in launch()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 353 def NCHWToNHWC(input_tensor): function
|