Lines Matching refs:data_layout
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()
144 … const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in compute_target()
145 … const int idx_height = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in compute_target()
150 …t = create_tensor<TensorType>(shape_scaled, _data_type, 1, _output_quantization_info, data_layout); in compute_target()
160 add_padding_x({ &src, &dst }, data_layout); in compute_target()
215 …pe, DataType data_type, QuantizationInfo quantization_info, DataLayout data_layout, InterpolationP… in setup() argument
221 data_layout, in setup()
235 …antization_info, QuantizationInfo output_quantization_info, DataLayout data_layout, InterpolationP… in setup() argument
242 data_layout, in setup()
256 …void setup(TensorShape shape, DataType data_type, DataLayout data_layout, InterpolationPolicy poli… in setup() argument
261 data_layout, in setup()