Searched refs:NCHWToNHWC (Results 1 - 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
H A Dpooling_ops_test.py47 def NCHWToNHWC(input_tensor): function
106 t = NCHWToNHWC(t)
368 actual_input_gradients = NCHWToNHWC(actual_input_gradients)
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dconv_ops_test.py214 conv = test_util.NCHWToNHWC(conv)
244 conv = test_util.NCHWToNHWC(conv)
297 expected = test_util.NCHWToNHWC(expected)
298 computed = test_util.NCHWToNHWC(computed)
546 conv = test_util.NCHWToNHWC(conv)
580 conv = test_util.NCHWToNHWC(conv)
906 conv_forward = test_util.NCHWToNHWC(conv_forward)
907 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_2)
960 conv_forward = test_util.NCHWToNHWC(conv_forward)
961 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_
[all...]
H A Dconv_ops_3d_test.py90 conv = test_util.NCHWToNHWC(conv)
345 conv = test_util.NCHWToNHWC(conv)
H A Dpooling_ops_3d_test.py83 t = test_util.NCHWToNHWC(t)
H A Ddepthtospace_op_test.py53 output_nhwc = test_util.NCHWToNHWC(output_nchw)
H A Dspacetodepth_op_test.py52 output_nhwc = test_util.NCHWToNHWC(output_nchw)
H A Dpooling_ops_test.py160 t = test_util.NCHWToNHWC(t)
942 t = test_util.NCHWToNHWC(t)
/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_pooling_gpu.cc109 functor::NCHWToNHWC<GPUDevice, T, 5>()(
239 functor::NCHWToNHWC<GPUDevice, T, 5>()(
H A Dconv_ops_gpu_3.cu.cc993 struct NCHWToNHWC<GPUDevice, T, NDIMS> { struct in namespace:tensorflow::functor
1052 template struct functor::NCHWToNHWC<GPUDevice, double, 4>; member in class:tensorflow::functor
1053 template struct functor::NCHWToNHWC<GPUDevice, float, 4>; member in class:tensorflow::functor
1054 template struct functor::NCHWToNHWC<GPUDevice, Eigen::half, 4>; member in class:tensorflow::functor
1070 template struct functor::NCHWToNHWC<GPUDevice, float, 5>; member in class:tensorflow::functor
1071 template struct functor::NCHWToNHWC<GPUDevice, Eigen::half, 5>; member in class:tensorflow::functor
H A Dpooling_ops_common.cc230 functor::NCHWToNHWC<GPUDevice, T, 4>()(
382 functor::NCHWToNHWC<GPUDevice, T, 4>()(
H A Dconv_2d.h260 struct NCHWToNHWC { struct in namespace:tensorflow::functor
H A Dfused_batch_norm_op.cc354 functor::NCHWToNHWC<GPUDevice, T, 4>()(
477 functor::NCHWToNHWC<GPUDevice, T, 4>()(
H A Dconv_ops_3d.cc459 functor::NCHWToNHWC<GPUDevice, T, 5>()(
H A Dconv_ops.cc773 functor::NCHWToNHWC<GPUDevice, T, 4>()(
H A Dconv_grad_input_ops.cc1045 functor::NCHWToNHWC<GPUDevice, T, 4>()(
H A Dconv_grad_ops_3d.cc746 functor::NCHWToNHWC<GPUDevice, T, 5>()(
/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
H A Dfused_conv2d_bias_activation_op_test.py222 output = test_util.NCHWToNHWC(output)
223 ref_output = test_util.NCHWToNHWC(ref_output)
262 output = test_util.NCHWToNHWC(output)
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
H A Dfused_conv2d_bias_activation_op.cc613 functor::NCHWToNHWC<GPUDevice, T, 4>()(
/external/tensorflow/tensorflow/python/framework/
H A Dtest_util.py298 def NCHWToNHWC(input_tensor): function

Completed in 182 milliseconds