Home
last modified time | relevance | path

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

123456

/external/ComputeLibrary/src/runtime/CL/functions/
DCLConvertFullyConnectedWeights.cpp33 DataLayout data_layout) in configure()
39 DataLayout data_layout) in configure()
47 DataLayout data_layout) in validate()
/external/ComputeLibrary/tests/validation/fixtures/
DUpsampleLayerFixture.h47 void setup(TensorShape input_shape, DataType data_type, DataLayout data_layout, in setup()
64 … DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in compute_target()
125 void setup(TensorShape input_shape, DataType data_type, DataLayout data_layout, in setup()
138 void setup(TensorShape input_shape, DataType data_type, DataLayout data_layout, in setup()
DNormalizePlanarYUVLayerFixture.h48 …void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout, Quantizati… in setup()
81 …rget(TensorShape shape0, const TensorShape &shape1, DataType dt, DataLayout data_layout, Quantizat… in compute_target()
146 void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout) in setup()
157 …void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout, Quantizati… in setup()
DScaleFixture.h48 …e shape, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, Interpola… in setup()
118 TensorType compute_target(TensorShape shape, DataLayout data_layout) in compute_target()
190 …e shape, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, Interpola… in setup()
208 …void setup(TensorShape shape, DataType data_type, DataLayout data_layout, InterpolationPolicy poli… in setup()
DPoolingLayerFixture.h80 DataType data_type, DataLayout data_layout, in compute_target()
121 …ute_reference(TensorShape shape, PoolingLayerInfo info, DataType data_type, DataLayout data_layout, in compute_reference()
142 …e, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout) in setup()
154 …e, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout) in setup()
DL2NormalizeLayerFixture.h52 … void setup(TensorShape shape, DataType data_type, DataLayout data_layout, int axis, float epsilon) in setup()
66 …TensorType compute_target(TensorShape shape, DataType data_type, DataLayout data_layout, int axis,… in compute_target()
100 … compute_reference(const TensorShape &shape, DataType data_type, DataLayout data_layout, int axis,… in compute_reference()
DYOLOLayerFixture.h50 …ivationFunction function, float alpha_beta, int32_t num_classes, DataLayout data_layout, DataType … in setup()
73 … in_place, const ActivationLayerInfo &info, int32_t num_classes, DataLayout data_layout, DataType … in compute_target()
142 …ivationFunction function, float alpha_beta, int32_t num_classes, DataLayout data_layout, DataType … in setup()
153 …ivationFunction function, float alpha_beta, int32_t num_classes, DataLayout data_layout, DataType … in setup()
DDirectConvolutionLayerFixture.h56 …ata_type, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout) in setup()
77 …ata_type, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout) in setup()
129 …e, QuantizationInfo quantization_info, ActivationLayerInfo act_info, const DataLayout &data_layout) in compute_target()
203 DataLayout data_layout) in setup()
216 ActivationLayerInfo act_info, DataLayout data_layout) in setup()
229 …ata_type, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout) in setup()
DNormalizationLayerFixture.h50 …e norm_type, int norm_size, float beta, bool is_scaled, DataType data_type, DataLayout data_layout) in setup()
66 …_target(TensorShape shape, NormalizationLayerInfo info, DataType data_type, DataLayout data_layout) in compute_target()
120 …e norm_type, int norm_size, float beta, bool is_scaled, DataType data_type, DataLayout data_layout) in setup()
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()
DDepthwiseConvolutionLayerFixture.h62 DataLayout data_layout, ActivationLayerInfo act_info) in setup()
121 const DataLayout data_layout, const ActivationLayerInfo &act_info) in compute_target()
194 …e_info, Size2D dilation, unsigned int depth_multiplier, DataType data_type, DataLayout data_layout, in setup()
209 DataLayout data_layout) in setup()
247 … unsigned int depth_multiplier, const DataType data_type, const DataLayout data_layout) in compute_target()
318 DataLayout data_layout, const ActivationLayerInfo &act_info, unsigned int n0) in setup()
362 …unsigned int depth_multiplier, const DataType data_type, const DataLayout data_layout, const Activ… in compute_target()
439 …ut_quantization_info, QuantizationInfo output_quantization_info, DataLayout data_layout, Activatio… in setup()
453 …ut_quantization_info, QuantizationInfo output_quantization_info, DataLayout data_layout, Activatio… in setup()
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()
DMaxUnpoolingLayerFixture.h50 …id setup(TensorShape shape, PoolingLayerInfo pool_info, DataType data_type, DataLayout data_layout) in setup()
80 DataType data_type, DataLayout data_layout, in compute_target()
149 …_type, Size2D pool_size, PadStrideInfo pad_stride_info, DataType data_type, DataLayout data_layout) in setup()
DDeconvolutionLayerFixture.h54 …DataType data_type, DataLayout data_layout, QuantizationInfo input_quantization_info, Quantization… in setup()
224 unsigned int num_kernels, DataType data_type, DataLayout data_layout, bool add_bias) in setup()
245 …ed int pad_bottom, unsigned int num_kernels, DataType data_type, DataLayout data_layout, bool add_… in setup()
266 …unsigned int num_kernels, DataType data_type, DataLayout data_layout, QuantizationInfo input_quant… in setup()
DChannelShuffleLayerFixture.h49 … void setup(TensorShape shape, unsigned int num_groups, DataType data_type, DataLayout data_layout) in setup()
62 …pute_target(TensorShape shape, DataType data_type, unsigned int num_groups, DataLayout data_layout) in compute_target()
DReorgLayerFixture.h47 void setup(TensorShape input_shape, int32_t stride, DataType data_type, DataLayout data_layout) in setup()
60 …compute_target(TensorShape input_shape, int32_t stride, DataType data_type, DataLayout data_layout) in compute_target()
DPriorBoxLayerFixture.h49 … setup(TensorShape input_shape, PriorBoxLayerInfo info, DataType data_type, DataLayout data_layout) in setup()
60 …hape, TensorShape output_shape, PriorBoxLayerInfo info, DataType data_type, DataLayout data_layout) in compute_target()
/external/tensorflow/tensorflow/python/kernel_tests/
Dbias_op_deterministic_test.py42 data_layout): argument
52 def _dataFormatFromDataLayout(self, data_layout=None): argument
78 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
/external/ComputeLibrary/src/core/CL/kernels/
DCLConvertFullyConnectedWeightsKernel.cpp43 DataLayout data_layout) in configure()
49 DataLayout data_layout) in configure()
92 DataLayout data_layout) in validate()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_bytesizeof_test.cc23 llvm::DataLayout data_layout( in TEST_F() local
32 llvm::DataLayout data_layout("e-m:e-i64:64-i128:128-n32:64-S128"); in TEST_F() local
/external/ComputeLibrary/tests/validation/reference/
DPoolingLayer.cpp41 …sor<T> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer_internal()
184 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer()
192 DataLayout data_layout) in pooling_layer()
201 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer()
210 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer()
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()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEConvertFullyConnectedWeights.cpp38 DataLayout data_layout) in configure()
45 DataLayout data_layout) in validate()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLConvertFullyConnectedWeights.h111 … configure(const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_layout) in configure()
122 …e_context, const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_layout) in configure()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h87 …output_shape(const TensorShape &input, size_t conv_w, size_t conv_h, const DataLayout &data_layout) in compute_vector_to_tensor_output_shape()
299 const auto data_layout = input.data_layout(); in compute_reshaped_depthwise_weights_shape() local
490 const DataLayout data_layout = input.data_layout(); in compute_deconvolution_upsampled_shape() local
524 const DataLayout data_layout = input.data_layout(); in compute_deconvolution_output_shape() local
561 const DataLayout data_layout = input->data_layout(); variable
708 const DataLayout data_layout = winograd_info.output_data_layout; in compute_winograd_output_transform_shape() local
1071 const DataLayout data_layout = input->data_layout(); in compute_batch_to_space_shape() local
1092 …ape compute_depth_to_space_shape(const TensorShape &input_shape, DataLayout data_layout, int block) in compute_depth_to_space_shape()
1155 const DataLayout data_layout = input->data_layout(); in compute_space_to_batch_shape() local
1181 const DataLayout data_layout = input->data_layout(); in compute_space_to_depth_shape() local
[all …]

123456