• Home
  • Raw
  • Download

Lines Matching refs:data_layout

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()
150 …e, QuantizationInfo quantization_info, ActivationLayerInfo act_info, const DataLayout &data_layout) in compute_target() argument
152 if(data_layout == DataLayout::NHWC) in compute_target()
160 …ype src = create_tensor<TensorType>(input_shape, data_type, 1, quantization_info, data_layout); in compute_target()
161 …e weights = create_tensor<TensorType>(weights_shape, data_type, 1, quantization_info, data_layout); in compute_target()
163 …pe dst = create_tensor<TensorType>(output_shape, data_type, 1, quantization_info, data_layout); in compute_target()
165 add_padding_x({ &src, &bias, &dst }, data_layout); in compute_target()
166 …add_padding_x({ &weights }, data_layout, input_shape[0] % 4 == 0); // Don't add left padding if cl… in compute_target()
235 DataLayout data_layout) in setup() argument
238 … act_info, data_layout, mixed_layout); in setup()
248 ActivationLayerInfo act_info, DataLayout data_layout) in setup() argument
251 … act_info, data_layout, mixed_layout); in setup()
261 …ata_type, QuantizationInfo quantization_info, ActivationLayerInfo act_info, DataLayout data_layout) in setup() argument
264 … act_info, data_layout); in setup()