| /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 | 120 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
259 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
|
| D | conv_ops_3d.cc | 527 functor::NCHWToNHWC<GPUDevice, T, 5>()( in launch()
562 void NCHWToNHWC<GPUDevice, T, 5>::operator()( \
|
| D | pooling_ops_common.cc | 413 functor::NCHWToNHWC<GPUDevice, RT, 4>()( in Compute()
753 functor::NCHWToNHWC<GPUDevice, T, 4>()( in Compute()
|
| D | lrn_op.cc | 289 functor::NCHWToNHWC<GPUDevice, T, 4>()( in launch()
619 functor::NCHWToNHWC<GPUDevice, T, 4>()( in launch()
|
| D | conv_grad_input_ops.cc | 430 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_2d.h | 460 struct NCHWToNHWC {
|
| D | conv_ops_fused_impl.h | 676 functor::NCHWToNHWC<GPUDevice, T, 4>()(
|
| D | fused_batch_norm_op.cc | 985 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
1167 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_2d_gpu.h | 1127 struct NCHWToNHWC<GPUDevice, T, NDIMS> {
|
| D | conv_ops.cc | 1153 functor::NCHWToNHWC<GPUDevice, T, 4>()( in operator ()()
|
| D | conv_grad_ops_3d.cc | 1566 functor::NCHWToNHWC<GPUDevice, T, 5>()( in Compute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 43 def NCHWToNHWC(input_tensor): function
102 t = NCHWToNHWC(t)
401 actual_input_gradients = NCHWToNHWC(actual_input_gradients)
412 actual_grad_gradients = NCHWToNHWC(actual_grad_gradients)
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | conv_ops_3d_test.py | 93 conv = test_util.NCHWToNHWC(conv)
162 expected = test_util.NCHWToNHWC(expected)
163 computed = test_util.NCHWToNHWC(computed)
541 conv = test_util.NCHWToNHWC(conv)
831 actual = test_util.NCHWToNHWC(actual)
832 expected = test_util.NCHWToNHWC(expected)
|
| D | conv_ops_test.py | 225 conv = test_util.NCHWToNHWC(conv)
255 conv = test_util.NCHWToNHWC(conv)
299 expected = test_util.NCHWToNHWC(expected)
300 computed = test_util.NCHWToNHWC(computed)
953 conv = test_util.NCHWToNHWC(conv)
987 conv = test_util.NCHWToNHWC(conv)
1348 conv_forward = test_util.NCHWToNHWC(conv_forward)
1349 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_2)
1394 conv_forward = test_util.NCHWToNHWC(conv_forward)
1395 conv_forward_2 = test_util.NCHWToNHWC(conv_forward_2)
[all …]
|
| D | pooling_ops_3d_test.py | 86 t = test_util.NCHWToNHWC(t)
|
| D | pooling_ops_test.py | 270 t = test_util.NCHWToNHWC(t)
1216 t = test_util.NCHWToNHWC(t)
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | SpaceToBatchNdTestImpl.cpp | 36 const armnn::PermutationVector NCHWToNHWC = {0, 3, 1, 2}; in SpaceToBatchNdTestImpl() local
39 inputTensorInfo = armnnUtils::Permuted(inputTensorInfo, NCHWToNHWC); in SpaceToBatchNdTestImpl()
40 outputTensorInfo = armnnUtils::Permuted(outputTensorInfo, NCHWToNHWC); in SpaceToBatchNdTestImpl()
43 armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, in SpaceToBatchNdTestImpl()
48 armnnUtils::Permute(outputTensorInfo.GetShape(), NCHWToNHWC, in SpaceToBatchNdTestImpl()
|
| D | L2NormalizationTestImpl.cpp | 44 const armnn::PermutationVector NCHWToNHWC = { 0, 3, 1, 2 }; in L2NormalizationTestImpl() local
49 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, inputData.data(), tmp.data(), sizeof(f… in L2NormalizationTestImpl()
62 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, expectedOutputValues.data(), tmp.data(… in L2NormalizationTestImpl()
|
| D | Pooling2dTestImpl.cpp | 308 const armnn::PermutationVector NCHWToNHWC = { 0, 3, 1, 2 }; in SimpleMaxPooling2dTestCommon() local
312 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, inputData.data(), tmp.data(), sizeof(T… in SimpleMaxPooling2dTestCommon()
316 …armnnUtils::Permute(outputTensorInfo.GetShape(), NCHWToNHWC, outputData.data(), tmp1.data(), sizeo… in SimpleMaxPooling2dTestCommon()
377 const armnn::PermutationVector NCHWToNHWC = { 0, 3, 1, 2 }; in SimpleAveragePooling2dTestCommon() local
381 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, inputData.data(), tmp.data(), sizeof(T… in SimpleAveragePooling2dTestCommon()
385 …armnnUtils::Permute(outputTensorInfo.GetShape(), NCHWToNHWC, outputData.data(), tmp1.data(), sizeo… in SimpleAveragePooling2dTestCommon()
486 const armnn::PermutationVector NCHWToNHWC = { 0, 3, 1, 2 }; in SimpleL2Pooling2dTestCommon() local
490 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, inputData.data(), tmp.data(), sizeof(T… in SimpleL2Pooling2dTestCommon()
494 …armnnUtils::Permute(outputTensorInfo.GetShape(), NCHWToNHWC, outputData.data(), tmp1.data(), sizeo… in SimpleL2Pooling2dTestCommon()
|
| D | Conv2dTestImpl.cpp | 272 const armnn::PermutationVector NCHWToNHWC = { 0, 3, 1, 2 }; in SimpleConvolution2dTestImpl() local
276 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, inputData.data(), tmp.data(), sizeof(T… in SimpleConvolution2dTestImpl()
306 …armnnUtils::Permute(outputTensorInfo.GetShape(), NCHWToNHWC, expectedOutput.data(), tmp.data(), si… in SimpleConvolution2dTestImpl()
321 …armnnUtils::Permute(kernelDesc.GetShape(), NCHWToNHWC, originalKernel.data(), kernel.data(), sizeo… in SimpleConvolution2dTestImpl()
1753 const armnn::PermutationVector NCHWToNHWC = { 0, 3, 1, 2 }; in DepthwiseConvolution2dAsymmetricTestImpl() local
1757 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, inputData.data(), tmp.data(), sizeof(T… in DepthwiseConvolution2dAsymmetricTestImpl()
1793 …armnnUtils::Permute(outputTensorInfo.GetShape(), NCHWToNHWC, outputData.data(), tmp.data(), sizeof… in DepthwiseConvolution2dAsymmetricTestImpl()
1907 const armnn::PermutationVector NCHWToNHWC = { 0, 3, 1, 2 }; in DepthwiseConvolution2dDepthMul1TestImpl() local
1911 …armnnUtils::Permute(inputTensorInfo.GetShape(), NCHWToNHWC, inputData.data(), tmp.data(), sizeof(T… in DepthwiseConvolution2dDepthMul1TestImpl()
1962 …armnnUtils::Permute(outputTensorInfo.GetShape(), NCHWToNHWC, outputImage.data(), tmp.data(), sizeo… in DepthwiseConvolution2dDepthMul1TestImpl()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | depthtospace_op_test.py | 48 output_nhwc = test_util.NCHWToNHWC(output_nchw)
63 output_nhwc = test_util.NCHWToNHWC(output_nchw)
|
| D | spacetodepth_op_test.py | 46 output_nhwc = test_util.NCHWToNHWC(output_nchw)
|