Home
last modified time | relevance | path

Searched refs:NHWCToNCHW (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dpooling_ops_test.py28 def NHWCToNCHW(input_tensor): function
93 t = NHWCToNCHW(t)
94 ksize = NHWCToNCHW(ksize)
95 strides = NHWCToNCHW(strides)
386 xla_inputs = NHWCToNCHW(inputs)
387 xla_outputs = NHWCToNCHW(outputs)
388 xla_output_gradients = NHWCToNCHW(output_gradients)
389 xla_output_grad_gradients = NHWCToNCHW(output_grad_gradients)
390 xla_ksize = NHWCToNCHW(ksize)
391 xla_strides = NHWCToNCHW(strides)
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d_gpu_double.cu.cc36 template struct NHWCToNCHW<Eigen::GpuDevice, double, 4>; variable
43 template struct NHWCToNCHW<Eigen::GpuDevice, double, 5>; variable
Dconv_2d_gpu_float.cu.cc45 template struct NHWCToNCHW<Eigen::GpuDevice, float, 4>; variable
52 template struct NHWCToNCHW<Eigen::GpuDevice, float, 5>; variable
Dconv_2d_gpu_half.cu.cc37 template struct NHWCToNCHW<Eigen::GpuDevice, Eigen::half, 4>; variable
44 template struct NHWCToNCHW<Eigen::GpuDevice, Eigen::half, 5>; variable
Dcudnn_pooling_gpu.cc55 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
185 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
190 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
194 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
Dpooling_ops_common.cc253 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
527 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
536 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
540 functor::NHWCToNCHW<GPUDevice, T, 4>()( in Compute()
Dlrn_op.cc245 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<GPUDevice>(), in launch()
539 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<GPUDevice>(), in launch()
549 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<GPUDevice>(), in launch()
559 functor::NHWCToNCHW<GPUDevice, T, 4>()( in launch()
Dconv_ops_3d.cc374 functor::NHWCToNCHW<GPUDevice, T, 5>()( in launch()
558 void NHWCToNCHW<GPUDevice, T, 5>::operator()( \
Dconv_grad_filter_ops.cc918 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
941 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
Dfused_batch_norm_op.cc877 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
1058 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
1070 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
Dconv_grad_input_ops.cc319 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
Dconv_2d.h450 struct NHWCToNCHW {
Dconv_grad_ops_3d.cc1464 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
1857 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
1876 functor::NHWCToNCHW<GPUDevice, T, 5>()( in Compute()
Dconv_ops_fused_impl.h447 functor::NHWCToNCHW<GPUDevice, T, 4>()(
Dconv_2d_gpu.h1109 struct NHWCToNCHW<GPUDevice, T, NDIMS> {
Dconv_ops.cc997 functor::NHWCToNCHW<GPUDevice, T, 4>()( in operator ()()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DSpaceToDepthTestImpl.cpp36 const armnn::PermutationVector NHWCToNCHW = {0, 2, 3, 1}; in SpaceToDepthTestImpl() local
40 inputTensorInfo = armnnUtils::Permuted(inputTensorInfo, NHWCToNCHW); in SpaceToDepthTestImpl()
41 outputTensorInfo = armnnUtils::Permuted(outputTensorInfo, NHWCToNCHW); in SpaceToDepthTestImpl()
44 armnnUtils::Permute(inputTensorInfo.GetShape(), NHWCToNCHW, in SpaceToDepthTestImpl()
49 armnnUtils::Permute(outputTensorInfo.GetShape(), NHWCToNCHW, in SpaceToDepthTestImpl()
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dpooling_ops_3d_test.py76 t = test_util.NHWCToNCHW(t)
77 window = test_util.NHWCToNCHW(window)
78 strides = test_util.NHWCToNCHW(strides)
293 ksize = test_util.NHWCToNCHW(ksize)
294 strides = test_util.NHWCToNCHW(strides)
295 t = test_util.NHWCToNCHW(t)
296 output_sizes = test_util.NHWCToNCHW(output_sizes)
Dconv_ops_test.py211 t1 = test_util.NHWCToNCHW(t1)
212 strides = test_util.NHWCToNCHW(strides)
213 dilations = test_util.NHWCToNCHW(dilations)
215 padding = test_util.NHWCToNCHW(padding)
250 t1 = test_util.NHWCToNCHW(t1)
251 strides = test_util.NHWCToNCHW(strides)
278 t1 = test_util.NHWCToNCHW(t1)
817 t1 = test_util.NHWCToNCHW(t1)
818 strides = test_util.NHWCToNCHW(strides)
819 dilations = test_util.NHWCToNCHW(dilations)
[all …]
Dconv_ops_3d_test.py88 t1 = test_util.NHWCToNCHW(t1)
89 strides = test_util.NHWCToNCHW(strides)
141 t1 = test_util.NHWCToNCHW(t1)
526 input_tensor = test_util.NHWCToNCHW(orig_input_tensor)
527 new_strides = test_util.NHWCToNCHW(strides)
808 input_sizes = test_util.NHWCToNCHW(input_sizes)
814 full_strides = test_util.NHWCToNCHW(full_strides)
815 full_dilations = test_util.NHWCToNCHW(full_dilations)
Dpooling_ops_test.py245 t = test_util.NHWCToNCHW(t)
246 ksize = test_util.NHWCToNCHW(ksize)
247 strides = test_util.NHWCToNCHW(strides)
249 padding = test_util.NHWCToNCHW(padding)
1202 padding = test_util.NHWCToNCHW(padding)
1203 t = test_util.NHWCToNCHW(input_tensor)
1279 t = test_util.NHWCToNCHW(input_tensor)
1292 t = test_util.NHWCToNCHW(t)
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Ddepthtospace_op_test.py45 input_nchw = test_util.NHWCToNCHW(input_nhwc)
60 input_nchw = test_util.NHWCToNCHW(input_nhwc)
Dspacetodepth_op_test.py43 input_nchw = test_util.NHWCToNCHW(input_nhwc)
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py369 def NHWCToNCHW(input_tensor): function