Searched refs:NHWCToNCHW (Results 1 - 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
H A Dpooling_ops_test.py32 def NHWCToNCHW(input_tensor): function
97 t = NHWCToNCHW(t)
98 ksize = NHWCToNCHW(ksize)
99 strides = NHWCToNCHW(strides)
354 xla_inputs = NHWCToNCHW(inputs)
355 xla_outputs = NHWCToNCHW(outputs)
356 xla_output_gradients = NHWCToNCHW(output_gradients)
357 xla_ksize = NHWCToNCHW(ksize)
358 xla_strides = NHWCToNCHW(strides)
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dpooling_ops_3d_test.py73 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)
H A Dconv_ops_test.py203 t1 = test_util.NHWCToNCHW(t1)
204 strides = test_util.NHWCToNCHW(strides)
205 dilations = test_util.NHWCToNCHW(dilations)
239 t1 = test_util.NHWCToNCHW(t1)
240 strides = test_util.NHWCToNCHW(strides)
276 t1 = test_util.NHWCToNCHW(t1)
535 input_sizes = test_util.NHWCToNCHW(input_sizes)
541 t2 = test_util.NHWCToNCHW(t2)
542 strides = test_util.NHWCToNCHW(strides)
562 new_input_sizes = test_util.NHWCToNCHW(input_size
[all...]
H A Dconv_ops_3d_test.py85 t1 = test_util.NHWCToNCHW(t1)
86 strides = test_util.NHWCToNCHW(strides)
330 input_tensor = test_util.NHWCToNCHW(orig_input_tensor)
331 new_strides = test_util.NHWCToNCHW(strides)
H A Dpooling_ops_test.py137 t = test_util.NHWCToNCHW(t)
138 ksize = test_util.NHWCToNCHW(ksize)
139 strides = test_util.NHWCToNCHW(strides)
929 t = test_util.NHWCToNCHW(input_tensor)
1005 t = test_util.NHWCToNCHW(input_tensor)
1018 t = test_util.NHWCToNCHW(t)
H A Ddepthtospace_op_test.py50 input_nchw = test_util.NHWCToNCHW(input_nhwc)
H A Dspacetodepth_op_test.py49 input_nchw = test_util.NHWCToNCHW(input_nhwc)
/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_pooling_gpu.cc53 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(),
175 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(),
180 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(),
184 functor::NHWCToNCHW<GPUDevice, T, 5>()(
H A Dpooling_ops_common.cc169 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(),
312 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(),
320 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(),
324 functor::NHWCToNCHW<GPUDevice, T, 4>()(
H A Dconv_ops_gpu_3.cu.cc975 struct NHWCToNCHW<GPUDevice, T, NDIMS> { struct in namespace:tensorflow::functor
1048 template struct functor::NHWCToNCHW<GPUDevice, double, 4>; member in class:tensorflow::functor
1049 template struct functor::NHWCToNCHW<GPUDevice, float, 4>; member in class:tensorflow::functor
1050 template struct functor::NHWCToNCHW<GPUDevice, Eigen::half, 4>; member in class:tensorflow::functor
1067 template struct functor::NHWCToNCHW<GPUDevice, float, 5>; member in class:tensorflow::functor
1068 template struct functor::NHWCToNCHW<GPUDevice, Eigen::half, 5>; member in class:tensorflow::functor
H A Dfused_batch_norm_op.cc264 functor::NHWCToNCHW<GPUDevice, T, 4>()(
404 functor::NHWCToNCHW<GPUDevice, T, 4>()(
416 functor::NHWCToNCHW<GPUDevice, T, 4>()(
H A Dconv_grad_ops_3d.cc611 functor::NHWCToNCHW<GPUDevice, T, 5>()(
967 functor::NHWCToNCHW<GPUDevice, T, 5>()(
985 functor::NHWCToNCHW<GPUDevice, T, 5>()(
H A Dconv_2d.h250 struct NHWCToNCHW { struct in namespace:tensorflow::functor
H A Dconv_ops_3d.cc299 functor::NHWCToNCHW<GPUDevice, T, 5>()(
H A Dconv_grad_filter_ops.cc843 functor::NHWCToNCHW<GPUDevice, T, 4>()(
864 functor::NHWCToNCHW<GPUDevice, T, 4>()(
H A Dconv_ops.cc617 functor::NHWCToNCHW<GPUDevice, T, 4>()(
H A Dconv_grad_input_ops.cc900 functor::NHWCToNCHW<GPUDevice, T, 4>()(
/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
H A Dfused_conv2d_bias_activation_op_test.py205 t1 = test_util.NHWCToNCHW(t1)
206 strides = test_util.NHWCToNCHW(strides)
250 t1 = test_util.NHWCToNCHW(t1)
251 strides = test_util.NHWCToNCHW(strides)
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
H A Dfused_conv2d_bias_activation_op.cc271 functor::NHWCToNCHW<GPUDevice, T, NDIMS>()(
/external/tensorflow/tensorflow/python/framework/
H A Dtest_util.py212 def NHWCToNCHW(input_tensor): function
/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dlayout_optimizer.cc791 bool NHWCToNCHW) {
808 if (NHWCToNCHW) {
788 AddNodeTranspose(const string& node_name, const string& input_name, const string& const_name, DataType data_type, const TensorShapeProto& input_shape, bool NHWCToNCHW) argument

Completed in 361 milliseconds