Searched refs:output_nhwc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | depthtospace_op_test.py | 48 output_nhwc = test_util.NCHWToNHWC(output_nchw)
52 self.evaluate(output_nhwc)
63 output_nhwc = test_util.NCHWToNHWC(output_nchw)
64 self.assertAllEqual(output_nhwc, outputs)
|
| D | spacetodepth_op_test.py | 46 output_nhwc = test_util.NCHWToNHWC(output_nchw)
47 self.assertAllEqual(self.evaluate(output_nhwc), outputs)
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuWinogradConv2d.cpp | 343 CpuAuxTensorHandler output_nhwc(offset_int_vec(PermutedOutput), _output_nhwc, tensors, true); in run() local
359 …{ ACL_SRC_0, winograd_output_transformed.get() }, { ACL_DST, is_nchw ? output_nhwc.get() : output … in run()
364 ITensorPack pack{ { ACL_SRC, output_nhwc.get() }, { ACL_DST, output } }; in run()
|