Home
last modified time | relevance | path

Searched defs:data_layout (Results 1 – 25 of 159) sorted by relevance

1234567

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DScaleFixture.h42 …e shape, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, Interpola… in setup()
68 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
133 TensorType compute_target(TensorShape shape, DataLayout data_layout) in compute_target()
215 …e shape, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, Interpola… in setup()
235 …ut_quantization_info, QuantizationInfo output_quantization_info, DataLayout data_layout, Interpola… in setup()
256 …void setup(TensorShape shape, DataType data_type, DataLayout data_layout, InterpolationPolicy poli… in setup()
H A DDirectConvolutionLayerFixture.h78 …ata_type, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout) in setup()
95 DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
150 …e, QuantizationInfo quantization_info, ActivationLayerInfo act_info, const DataLayout &data_layout) in compute_target()
235 DataLayout data_layout) in setup()
248 ActivationLayerInfo act_info, DataLayout data_layout) in setup()
261 …ata_type, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout) in setup()
H A DPoolingLayerFixture.h61 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
94 DataType data_type, DataLayout data_layout, in compute_target()
143 …ute_reference(TensorShape shape, PoolingLayerInfo info, DataType data_type, DataLayout data_layout, in compute_reference()
165 …e, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout) in setup()
177 …e, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout) in setup()
H A DWinogradConvolutionLayerFixture.h60 DataType data_type, ActivationLayerInfo act_info, const DataLayout &data_layout) in setup()
72 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
110 … DataType data_type, ActivationLayerInfo act_info, const DataLayout data_layout) in compute_target()
247 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d… in setup()
297 …st TensorShape &output_shape, const WinogradInfo &winograd_info, DataLayout data_layout, DataType … in compute_target()
359 …void setup(TensorShape input_shape, Size2D output_tile, DataLayout data_layout, DataType data_type) in setup()
411 …st TensorShape &output_shape, const WinogradInfo &winograd_info, DataLayout data_layout, DataType … in compute_target()
H A DNormalizePlanarYUVLayerFixture.h48 …void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout, Quantizati… in setup()
83 …rget(TensorShape shape0, const TensorShape &shape1, DataType dt, DataLayout data_layout, Quantizat… in compute_target()
148 void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout) in setup()
159 …void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout, Quantizati… in setup()
H A DDepthwiseConvolutionLayerFixture.h293 …e_info, Size2D dilation, unsigned int depth_multiplier, DataType data_type, DataLayout data_layout, in setup()
308 DataLayout data_layout) in setup()
447 …DataLayout data_layout, const ActivationLayerInfo &act_info, unsigned int n0, bool export_to_cl_im… in setup()
645 …ut_quantization_info, QuantizationInfo output_quantization_info, DataLayout data_layout, Activatio… in setup()
659 …ut_quantization_info, QuantizationInfo output_quantization_info, DataLayout data_layout, Activatio… in setup()
H A DDeconvolutionLayerFixture.h54 DataType data_type, DataType weights_data_type, DataLayout data_layout, in setup()
251 unsigned int num_kernels, DataType data_type, DataLayout data_layout, bool add_bias) in setup()
272 …ed int pad_bottom, unsigned int num_kernels, DataType data_type, DataLayout data_layout, bool add_… in setup()
293 …unsigned int num_kernels, DataType data_type, DataLayout data_layout, QuantizationInfo input_quant… in setup()
315 …unsigned int num_kernels, DataType data_type, DataLayout data_layout, QuantizationInfo input_quant… in setup()
H A DSpaceToBatchFixture.h44 DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in setup()
64 … DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in compute_target()
145 DataType data_type, DataLayout data_layout) in setup()
157 DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in setup()
H A DROIPoolingLayerFixture.h49 …LayerInfo pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout, Quantizat… in setup()
106 DataLayout data_layout, in compute_target()
179 DataLayout data_layout, QuantizationInfo qinfo, QuantizationInfo output_qinfo) in setup()
191 … ROIPoolingLayerInfo pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout) in setup()
H A DROIAlignLayerFixture.h49 …LayerInfo pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout, Quantizat… in setup()
117 DataLayout data_layout, in compute_target()
193 … ROIPoolingLayerInfo pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout) in setup()
206 DataLayout data_layout, QuantizationInfo qinfo, QuantizationInfo output_qinfo) in setup()
H A DConvolutionLayerFixture.h367 DataLayout data_layout, ActivationLayerInfo act_info) in setup()
395 … DataLayout data_layout, QuantizationInfo quantization_info, ActivationLayerInfo act_info) in setup()
408 …DataLayout data_layout, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataType… in setup()
426 const DataLayout data_layout = tensor_info.data_layout(); in prepare_weights() local
465 const DataLayout data_layout = src.data_layout(); in rearrange_data() local
518 … bias_shape, TensorShape output_shape, PadStrideInfo info, Size2D dilation, DataLayout data_layout, in setup()
H A DL2NormalizeLayerFixture.h52 … void setup(TensorShape shape, DataType data_type, DataLayout data_layout, int axis, float epsilon) in setup()
69 …TensorType compute_target(TensorShape shape, DataType data_type, DataLayout data_layout, int axis,… in compute_target()
103 … compute_reference(const TensorShape &shape, DataType data_type, DataLayout data_layout, int axis,… in compute_reference()
H A DNormalizationLayerFixture.h50 …e norm_type, int norm_size, float beta, bool is_scaled, DataType data_type, DataLayout data_layout) in setup()
69 …_target(TensorShape shape, NormalizationLayerInfo info, DataType data_type, DataLayout data_layout) in compute_target()
123 …e norm_type, int norm_size, float beta, bool is_scaled, DataType data_type, DataLayout data_layout) in setup()
H A DMaxUnpoolingLayerFixture.h50 …id setup(TensorShape shape, PoolingLayerInfo pool_info, DataType data_type, DataLayout data_layout) in setup()
85 DataType data_type, DataLayout data_layout, in compute_target()
154 …_type, Size2D pool_size, PadStrideInfo pad_stride_info, DataType data_type, DataLayout data_layout) in setup()
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DConvolutionLayer.cpp97 …rShape &weights_shape, const TensorShape &output_shape, DataType data_type, DataLayout data_layout, in is_post_op_list_valid_in_gemmconv()
293 const auto data_layout = DataLayout::NHWC; in TEST_SUITE() local
320 const auto data_layout = DataLayout::NCHW; in TEST_CASE() local
341 const auto data_layout = DataLayout::NHWC; in TEST_CASE() local
362 const auto data_layout = DataLayout::NHWC; in TEST_CASE() local
383 const auto data_layout = DataLayout::NHWC; in TEST_CASE() local
406 const auto data_layout = DataLayout::NHWC; in TEST_SUITE() local
420 const auto data_layout = DataLayout::NHWC; in TEST_CASE() local
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h88 …output_shape(const TensorShape &input, size_t conv_w, size_t conv_h, const DataLayout &data_layout) in compute_vector_to_tensor_output_shape()
426 const DataLayout data_layout = input.data_layout(); in compute_depthwise_convolution_shape() local
464 const DataLayout data_layout = input.data_layout(); in compute_deconvolution_upsampled_shape() local
498 const DataLayout data_layout = input.data_layout(); in compute_deconvolution_output_shape() local
535 const DataLayout data_layout = input->data_layout(); variable
682 const DataLayout data_layout = winograd_info.output_data_layout; in compute_winograd_output_transform_shape() local
1085 const DataLayout data_layout = input->data_layout(); in compute_batch_to_space_shape() local
1106 …ape compute_depth_to_space_shape(const TensorShape &input_shape, DataLayout data_layout, int block) in compute_depth_to_space_shape()
1169 const DataLayout data_layout = input->data_layout(); in compute_space_to_batch_shape() local
1195 const DataLayout data_layout = input->data_layout(); in compute_space_to_depth_shape() local
[all …]
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DPoolingLayer.cpp41 …sor<T> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer_internal()
183 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer()
191 DataLayout data_layout) in pooling_layer()
200 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer()
209 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer()
H A DMaxUnpoolingLayer.cpp43 TensorShape output_shape, DataLayout data_layout) in max_unpooling_layer_internal()
82 TensorShape output_shape, DataLayout data_layout) in max_unpooling_layer()
94 TensorShape output_shape, DataLayout data_layout) in max_unpooling_layer()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
H A DDirectConv2dFixture.h93 …DataLayout data_layout, QuantizationInfo quantization_info, QuantizationInfo weight_quantization_i… in setup()
208 …adStrideInfo &info, const Size2D &dialation, DataType data_type, DataLayout data_layout, Quantizat… in setup()
227 DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout) in setup()
249 …a_type, DataType bias_data_type, QuantizationInfo quantization_info, const DataLayout &data_layout) in compute_target()
346 DataLayout data_layout) in setup()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLConvertFullyConnectedWeights.cpp50 DataLayout data_layout) in configure()
56 DataLayout data_layout) in configure()
67 DataLayout data_layout) in validate()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuPool2dKernel.cpp159 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_lay… in validate_arguments() local
218 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in validate_and_configure_window() local
287 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in configure() local
335 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : pool_… in validate() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dbias_op_d9m_test.py38 data_layout): argument
48 def _dataFormatFromDataLayout(self, data_layout=None): argument
74 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/
H A DResizeFixture.h56 …(TensorShape shape, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, in setup()
227 …void setup(TensorShape shape, DataType data_type, DataLayout data_layout, InterpolationPolicy poli… in setup()
245 …e shape, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, Interpola… in setup()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dpooling_ops_common.cc269 se::dnn::DataLayout data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() local
273 se::dnn::DataLayout data_layout; in Compute() local
544 se::dnn::DataLayout data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() local
564 se::dnn::DataLayout data_layout; in Compute() local
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLDeconvolutionReshapeOutputKernel.cpp45 const DataLayout data_layout = input_info->data_layout(); in validate_arguments() local
95 const DataLayout data_layout = input_info->data_layout(); in validate_and_configure_window() local
137 const DataLayout data_layout = input_info->data_layout(); in configure() local

1234567