| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | PoolingLayerFixture.h | 49 …hape shape, PoolingLayerInfo pool_info, DataType data_type, DataLayout data_layout, bool indices =…
54 …_target = compute_target(shape, pool_info, data_type, data_layout, input_qinfo, output_qinfo…
55 …_reference = compute_reference(shape, pool_info, data_type, data_layout, input_qinfo, output_qi…
61 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
63 …src.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
64 …dst.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
70 src.info()->set_data_layout(data_layout); in mix_layout()
71 dst.info()->set_data_layout(data_layout); in mix_layout()
94 DataType data_type, DataLayout data_layout, in compute_target() argument
99 if(data_layout == DataLayout::NHWC) in compute_target()
[all …]
|
| D | ScaleFixture.h | 42 …pe, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, InterpolationP… in setup() argument
61 _target = compute_target(shape, data_layout); in setup()
68 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
70 …src.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
71 …dst.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
77 src.info()->set_data_layout(data_layout); in mix_layout()
78 dst.info()->set_data_layout(data_layout); in mix_layout()
133 TensorType compute_target(TensorShape shape, DataLayout data_layout) in compute_target() argument
136 if(data_layout == DataLayout::NHWC) in compute_target()
142 …rType src = create_tensor<TensorType>(shape, _data_type, 1, _input_quantization_info, data_layout); in compute_target()
[all …]
|
| D | DirectConvolutionLayerFixture.h | 56 …zationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout, bool mixed_lay…
72 …as_shape, output_shape, info, data_type, bias_data_type, quantization_info, act_info, data_layout);
78 …ata_type, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout) in setup() argument
80 ARM_COMPUTE_ERROR_ON(data_layout == DataLayout::UNKNOWN); in setup()
88 …as_shape, output_shape, info, data_type, bias_data_type, quantization_info, act_info, data_layout); in setup()
95 DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
97 …src.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
98 …dst.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
104 src.info()->set_data_layout(data_layout); in mix_layout()
105 dst.info()->set_data_layout(data_layout); in mix_layout()
[all …]
|
| D | WinogradConvolutionLayerFixture.h | 60 DataType data_type, ActivationLayerInfo act_info, const DataLayout &data_layout) in setup() argument
65 …rget(input_shape, weights_shape, bias_shape, output_shape, info, data_type, act_info, data_layout); in setup()
72 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
74 …src.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
75 …dst.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
81 src.info()->set_data_layout(data_layout); in mix_layout()
82 dst.info()->set_data_layout(data_layout); in mix_layout()
110 … DataType data_type, ActivationLayerInfo act_info, const DataLayout data_layout) in compute_target() argument
112 if(data_layout == DataLayout::NHWC) in compute_target()
120 …pe src = create_tensor<TensorType>(input_shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
[all …]
|
| D | L2NormalizeLayerFixture.h | 52 … void setup(TensorShape shape, DataType data_type, DataLayout data_layout, int axis, float epsilon) in setup() argument
54 _target = compute_target(shape, data_type, data_layout, axis, epsilon); in setup()
55 _reference = compute_reference(shape, data_type, data_layout, axis, epsilon); in setup()
69 …TensorType compute_target(TensorShape shape, DataType data_type, DataLayout data_layout, int axis,… in compute_target() argument
71 if(data_layout == DataLayout::NHWC) in compute_target()
77 … TensorType src = create_tensor<TensorType>(shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
78 … TensorType dst = create_tensor<TensorType>(shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
103 …ute_reference(const TensorShape &shape, DataType data_type, DataLayout data_layout, int axis, floa… in compute_reference() argument
106 if(data_layout == DataLayout::NHWC) in compute_reference()
|
| D | ROIPoolingLayerFixture.h | 49 …Info pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout, QuantizationIn… in setup() argument
51 …_target = compute_target(input_shape, data_type, data_layout, pool_info, rois_shape, qinfo, out… in setup()
63 …nst ROIPoolingLayerInfo &pool_info, TensorShape rois_shape, DataLayout data_layout = DataLayout::N…
76 …led_width = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
77 …led_height = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
106 DataLayout data_layout, in compute_target() argument
115 … TensorType src = create_tensor<TensorType>(input_shape, data_type, 1, qinfo, data_layout); in compute_target()
120 …sorType dst = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
141 generate_rois(AccessorType(rois_tensor), input_shape, pool_info, rois_shape, data_layout); in compute_target()
179 DataLayout data_layout, QuantizationInfo qinfo, QuantizationInfo output_qinfo) in setup() argument
[all …]
|
| D | SpaceToBatchFixture.h | 44 DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in setup() argument
46 …put_shape, block_shape_shape, paddings_shape, output_shape, data_type, data_layout, quantization_i… in setup()
64 … DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in compute_target() argument
66 if(data_layout == DataLayout::NHWC) in compute_target()
73 …input = create_tensor<TensorType>(input_shape, data_type, 1, quantization_info, data_layout); in compute_target()
76 …utput = create_tensor<TensorType>(output_shape, data_type, 1, quantization_info, data_layout); in compute_target()
103 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WID… in compute_target()
145 DataType data_type, DataLayout data_layout) in setup() argument
147 …put_shape, block_shape_shape, paddings_shape, output_shape, data_type, data_layout, QuantizationIn… in setup()
157 DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in setup() argument
[all …]
|
| D | MaxUnpoolingLayerFixture.h | 50 …id setup(TensorShape shape, PoolingLayerInfo pool_info, DataType data_type, DataLayout data_layout) in setup() argument
60 … _target = compute_target(shape, pool_info, data_type, data_layout, input_qinfo, output_qinfo); in setup()
85 DataType data_type, DataLayout data_layout, in compute_target() argument
89 if(data_layout == DataLayout::NHWC) in compute_target()
95 … src = create_tensor<TensorType>(input_shape, data_type, 1, input_qinfo, data_layout); in compute_target()
97 …e dst = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
98 … unpooled = create_tensor<TensorType>(input_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
99 … indices = create_tensor<TensorType>(dst_shape, DataType::U32, 1, output_qinfo, data_layout); in compute_target()
154 …_type, Size2D pool_size, PadStrideInfo pad_stride_info, DataType data_type, DataLayout data_layout) in setup() argument
156 …rType, F1, F2, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_inf… in setup()
[all …]
|
| D | ROIAlignLayerFixture.h | 49 …Info pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout, QuantizationIn… in setup() argument
52 …_target = compute_target(input_shape, data_type, data_layout, pool_info, rois_shape, qinfo… in setup()
64 …nst ROIPoolingLayerInfo &pool_info, TensorShape rois_shape, DataLayout data_layout = DataLayout::N…
77 …led_width = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
78 …led_height = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
117 DataLayout data_layout, in compute_target() argument
123 if(data_layout == DataLayout::NHWC) in compute_target()
131 … TensorType src = create_tensor<TensorType>(input_shape, data_type, 1, qinfo, data_layout); in compute_target()
135 …e dst = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
156 generate_rois(AccessorType(rois_tensor), input_shape, pool_info, rois_shape, data_layout); in compute_target()
[all …]
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 88 …output_shape(const TensorShape &input, size_t conv_w, size_t conv_h, const DataLayout &data_layout) in compute_vector_to_tensor_output_shape() argument
90 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in compute_vector_to_tensor_output_shape()
91 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in compute_vector_to_tensor_output_shape()
92 const size_t idx_c = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in compute_vector_to_tensor_output_shape()
125 …const size_t idx_width = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimensio… in compute_reorg_output_shape()
126 …const size_t idx_height = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimensio… in compute_reorg_output_shape()
127 …const size_t idx_channel = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimensio… in compute_reorg_output_shape()
154 ARM_COMPUTE_ERROR_ON(weights.data_layout() == DataLayout::NHWC && num_groups > 1);
378 const DataLayout data_layout = input.data_layout();
379 …const int width_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W…
[all …]
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuPool2dKernel.cpp | 159 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_lay… in validate_arguments() local
160 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_arguments()
161 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_arguments()
184 && (src->data_layout() == DataLayout::NHWC), in validate_arguments()
199 …get_implementation(PoolDataTypeISASelectorData{ src->data_type(), src->data_layout(), pool_stride_… in validate_arguments()
218 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in validate_and_configure_window() local
222 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_and_configure_window()
223 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimen… in validate_and_configure_window()
287 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in configure() local
288 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure()
[all …]
|
| D | CpuPool3dKernel.cpp | 71 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(src->data_layout() != DataLayout::NDHWC, "Only NDHWC layout suppor… in validate_arguments()
79 const auto data_layout = src->data_layout(); in validate_arguments() local
80 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
81 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
82 … const int idx_depth = get_data_layout_dimension_index(data_layout, DataLayoutDimension::DEPTH); in validate_arguments()
132 const auto data_layout = src->data_layout(); in configure() local
133 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure()
134 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure()
135 … const int idx_depth = get_data_layout_dimension_index(data_layout, DataLayoutDimension::DEPTH); in configure()
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | bias_op_d9m_test.py | 38 data_layout): argument
40 if data_layout == 'channels_first':
42 elif data_layout == 'channels_last':
48 def _dataFormatFromDataLayout(self, data_layout=None): argument
49 if data_layout == 'channels_first':
51 elif data_layout == 'channels_last':
69 data_layout=['channels_first', 'channels_last'],
74 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
81 hash(data_layout) % 256 + hash(data_rank) % 256 +
88 data_dim, data_layout)
[all …]
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLDeconvolutionReshapeOutputKernel.cpp | 45 const DataLayout data_layout = input_info->data_layout(); in validate_arguments() local
47 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
48 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
49 const size_t idx_b = get_data_layout_dimension_index(data_layout, DataLayoutDimension::BATCHES); in validate_arguments()
95 const DataLayout data_layout = input_info->data_layout(); in validate_and_configure_window() local
96 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension… in validate_and_configure_window()
97 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension… in validate_and_configure_window()
104 …output, input->clone()->set_tensor_shape(output_shape).set_data_layout(data_layout).set_quantizati… in validate_and_configure_window()
137 const DataLayout data_layout = input_info->data_layout(); in configure() local
138 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in configure()
[all …]
|
| D | CLNormalizationLayerKernel.cpp | 71 const DataLayout data_layout = input->data_layout(); in validate_and_configure_window() local
72 if(data_layout == DataLayout::NCHW) in validate_and_configure_window()
75 …gned int norm_idx = get_normalization_dimension_index(input->data_layout(), norm_info); in validate_and_configure_window()
153 const DataLayout data_layout = input->info()->data_layout(); in configure() local
156 if(norm_info.is_cross_map() && data_layout == DataLayout::NHWC) in configure()
162 if(data_layout == DataLayout::NCHW) in configure()
164 const unsigned int norm_idx = get_normalization_dimension_index(data_layout, norm_info); in configure()
194 …build_opts.add_option_if(norm_info.is_in_map() || (data_layout == DataLayout::NHWC && norm_info.is… in configure()
195 …build_opts.add_option_if(norm_info.is_in_map() && data_layout == DataLayout::NHWC, "-DDIM1_SIZE=" … in configure()
201 … kernel_name = "normalization_layer_in_map_" + lower_string(string_from_data_layout(data_layout)); in configure()
[all …]
|
| D | CLSpaceToBatchLayerKernel.cpp | 54 const DataLayout data_layout = input->data_layout(); in validate_arguments() local
55 …const int idx_channel = get_data_layout_dimension_index(data_layout, DataLayoutDimension::C… in validate_arguments()
106 const DataLayout data_layout = input->info()->data_layout(); in configure() local
107 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in configure()
108 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in configure()
109 …const int idx_batch = get_data_layout_dimension_index(data_layout, DataLayoutDimension::B… in configure()
120 …space_to_batch_" + lower_string(string_from_data_layout(input->info()->data_layout())), build_opts… in configure()
148 const DataLayout data_layout = input->info()->data_layout(); in configure() local
149 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in configure()
150 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in configure()
[all …]
|
| D | CLDepthToSpaceLayerKernel.cpp | 46 const DataLayout data_layout = input->data_layout(); in validate_arguments() local
47 …const int idx_channel = get_data_layout_dimension_index(data_layout, DataLayoutDimension::C… in validate_arguments()
53 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments()
54 … const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments()
80 …ute_depth_to_space_shape(input->info()->tensor_shape(), input->info()->data_layout(), block_shape); in configure()
91 …const int idx_width = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDi… in configure()
92 …const int idx_channel = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDi… in configure()
100 …depth_to_space_" + lower_string(string_from_data_layout(input->info()->data_layout())), build_opts… in configure()
|
| D | CLSpaceToDepthLayerKernel.cpp | 49 const DataLayout data_layout = input->data_layout(); in validate_arguments() local
50 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in validate_arguments()
51 …const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in validate_arguments()
52 …const int idx_channel = get_data_layout_dimension_index(data_layout, DataLayoutDimension::C… in validate_arguments()
53 …const int idx_batch = get_data_layout_dimension_index(data_layout, DataLayoutDimension::B… in validate_arguments()
91 …const int idx_width = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDi… in configure()
92 …const int idx_channel = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDi… in configure()
100 …space_to_depth_" + lower_string(string_from_data_layout(input->info()->data_layout())), build_opts… in configure()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_bytesizeof_test.cc | 23 llvm::DataLayout data_layout( in TEST_F() local
27 EXPECT_EQ(xla::llvm_ir::ByteSizeOf(tuple_shape, data_layout), in TEST_F()
28 data_layout.getPointerSize(0 /* default address space */)); in TEST_F()
32 llvm::DataLayout data_layout("e-m:e-i64:64-i128:128-n32:64-S128"); in TEST_F() local
35 EXPECT_EQ(xla::llvm_ir::ByteSizeOf(tuple_shape, data_layout), in TEST_F()
36 data_layout.getPointerSize(0 /* default address space */)); in TEST_F()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLConvertFullyConnectedWeights.cpp | 50 DataLayout data_layout) in configure() argument
52 …re(CLKernelLibrary::get().get_compile_context(), input, output, original_input_shape, data_layout); in configure()
56 DataLayout data_layout) in configure() argument
59 ARM_COMPUTE_LOG_PARAMS(input, output, original_input_shape, data_layout); in configure()
63 …figure(compile_context, _impl->src->info(), _impl->dst->info(), original_input_shape, data_layout); in configure()
67 DataLayout data_layout) in validate() argument
69 …opencl::ClConvertFullyConnectedWeights::validate(input, output, original_input_shape, data_layout); in validate()
|
| D | CLDirectDeconvolutionLayer.cpp | 64 const DataLayout data_layout = input->data_layout(); in validate() local
66 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate()
67 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate()
68 const size_t idx_c = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); in validate()
106 …is_resizable(true).reset_padding().set_tensor_shape(scale_out_shape).set_data_layout(data_layout)); in validate()
134 const DataLayout data_layout = input->info()->data_layout(); in configure() local
136 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure()
137 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure()
141 _weights_flipped.allocator()->init(weights->info()->clone()->set_data_layout(data_layout)); in configure()
149 …tput->info(), input->info()->clone()->set_tensor_shape(output_shape).set_data_layout(data_layout)); in configure()
[all …]
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLConvertFullyConnectedWeights.h | 74 …Tensor *input, ICLTensor *output, const TensorShape &original_input_shape, DataLayout data_layout);
85 …Tensor *input, ICLTensor *output, const TensorShape &original_input_shape, DataLayout data_layout);
93 …input, const ITensorInfo *output, const TensorShape &original_input_shape, DataLayout data_layout);
140 … configure(const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_layout) in configure() argument
142 … configure(CLKernelLibrary::get().get_compile_context(), input, original_input_shape, data_layout); in configure()
151 …e_context, const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_layout) in configure() argument
153 _func.configure(compile_context, input, &_output, original_input_shape, data_layout); in configure()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | PoolingLayer.cpp | 41 …sor<T> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer_internal() argument
100 if(data_layout == DataLayout::NCHW) in pooling_layer_internal()
176 …loat> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout);
178 …half> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout);
180 …half> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout);
183 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer() argument
186 return pooling_layer_internal<T, T>(src, info, indices, data_layout); in pooling_layer()
191 DataLayout data_layout) in pooling_layer() argument
194 …SimpleTensor<float> dst_tmp = pooling_layer_internal<float>(src_tmp, info, indices, data_layout); in pooling_layer()
200 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer() argument
[all …]
|
| D | MaxUnpoolingLayer.cpp | 43 TensorShape output_shape, DataLayout data_layout) in max_unpooling_layer_internal() argument
47 ARM_COMPUTE_UNUSED(data_layout); in max_unpooling_layer_internal()
82 TensorShape output_shape, DataLayout data_layout) in max_unpooling_layer() argument
86 …x_unpooling_layer_internal<float>(src_tmp, info, output_qinfo, indices, output_shape, data_layout); in max_unpooling_layer()
94 TensorShape output_shape, DataLayout data_layout) in max_unpooling_layer() argument
96 …eturn max_unpooling_layer_internal<T>(src, info, output_qinfo, indices, output_shape, data_layout); in max_unpooling_layer()
101 TensorShape output_shape, DataLayout data_layout);
104 TensorShape output_shape, DataLayout data_layout);
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClConvertFullyConnectedWeights.cpp | 35 … ITensorInfo *src, ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout) in configure() argument
37 ARM_COMPUTE_LOG_PARAMS(src, dst, original_src_shape, data_layout); in configure()
39 k->configure(compile_context, src, dst, original_src_shape, data_layout); in configure()
43 …orInfo *src, const ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout) in validate() argument
45 …kernels::ClConvertFullyConnectedWeightsKernel::validate(src, dst, original_src_shape, data_layout); in validate()
|