Home
last modified time | relevance | path

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

1234567

/external/ComputeLibrary/tests/validation/fixtures/
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
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()
DDepthwiseConvolutionLayerFixture.h293 …e_info, Size2D dilation, unsigned int depth_multiplier, DataType data_type, DataLayout data_layout, in setup()
308 DataLayout data_layout) in setup()
447DataLayout 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()
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()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLConvertFullyConnectedWeights.cpp50 DataLayout data_layout) in configure()
56 DataLayout data_layout) in configure()
67 DataLayout data_layout) in validate()
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
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
/external/ComputeLibrary/tests/validation/CL/
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
/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()
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()
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/gpu/cl/operators/
DClConvertFullyConnectedWeights.cpp35 … ITensorInfo *src, ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout) in configure()
43 …orInfo *src, const ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout) in validate()
/external/ComputeLibrary/src/cpu/operators/
DCpuConvertFullyConnectedWeights.cpp34 … ITensorInfo *src, ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout) in configure()
42 …orInfo *src, const ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout) in validate()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLConvertFullyConnectedWeights.h140 … configure(const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_layout) in configure()
151 …e_context, const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_layout) in configure()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEConvertFullyConnectedWeights.cpp44 DataLayout data_layout) in configure()
55 DataLayout data_layout) in validate()

1234567