Home
last modified time | relevance | path

Searched refs:input_nhwc (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/
DCpuWinogradConv2dKernel.cpp39 const ITensor *input_nhwc = tensors.get_const_tensor(TensorType::ACL_SRC); in run_op() local
46 int element_size_in_bytes = input_nhwc->info()->element_size(); in run_op()
47 const auto src_strides = input_nhwc->info()->strides_in_bytes(); in run_op()
52 … auto input_nhwc_ptr = reinterpret_cast<const void *>(input_nhwc->buffer() + input_nhwc->inf… in run_op()
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Ddepthtospace_op_test.py37 input_nhwc = math_ops.cast(inputs, dtype)
40 x_tf = array_ops.depth_to_space(input_nhwc, block_size)
45 input_nchw = test_util.NHWCToNCHW(input_nhwc)
57 x_tf = array_ops.depth_to_space(input_nhwc, block_size)
60 input_nchw = test_util.NHWCToNCHW(input_nhwc)
116 input_nhwc = array_ops.ones([batch_size, 2, 3, 12])
121 x_tf = array_ops.depth_to_space(input_nhwc, block_size)
127 x_tf = array_ops.depth_to_space(input_nhwc, block_size)
Dspacetodepth_op_test.py35 input_nhwc = math_ops.cast(inputs, dtype)
37 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
43 input_nchw = test_util.NHWCToNCHW(input_nhwc)
/external/ComputeLibrary/src/cpu/operators/
DCpuWinogradConv2d.cpp330 CpuAuxTensorHandler input_nhwc(offset_int_vec(PermutedInput), _input_nhwc, tensors, true); in run() local
337 ITensorPack pack{ { ACL_SRC, src }, { ACL_DST, input_nhwc.get() } }; in run()
345 …ITensorPack transform_input_pack{ { ACL_SRC, is_nchw ? input_nhwc.get() : src }, { ACL_DST, winogr… in run()