• Home
  • Raw
  • Download

Lines Matching refs:data_layout

60                DataType data_type, ActivationLayerInfo act_info, const DataLayout &data_layout)  in setup()  argument
65 …rget(input_shape, weights_shape, bias_shape, output_shape, info, data_type, act_info, data_layout); in setup()
72 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
74 …src.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
75 …dst.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); in mix_layout()
81 src.info()->set_data_layout(data_layout); in mix_layout()
82 dst.info()->set_data_layout(data_layout); in mix_layout()
110 … DataType data_type, ActivationLayerInfo act_info, const DataLayout data_layout) in compute_target() argument
112 if(data_layout == DataLayout::NHWC) in compute_target()
120 …pe src = create_tensor<TensorType>(input_shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
121 … weights = create_tensor<TensorType>(weights_shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
122 …ype bias = create_tensor<TensorType>(bias_shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
123 …e dst = create_tensor<TensorType>(output_shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
136 add_padding_x({ &src, &weights, &bias, &dst }, data_layout); in compute_target()
215 src_t1.data_layout()); in compute_reference()
247 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d… in setup() argument
251 … = compute_target(input_shape, output_shape, winograd_info, data_layout, data_type); in setup()
258 const DataLayout data_layout_src = src.info()->data_layout(); in mix_layout()
259 const DataLayout data_layout_dst = dst.info()->data_layout(); in mix_layout()
297 …nsorShape &output_shape, const WinogradInfo &winograd_info, DataLayout data_layout, DataType data_… in compute_target() argument
299 if(data_layout == DataLayout::NHWC) in compute_target()
304 …orType src = create_tensor<TensorType>(input_shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
314 add_padding_x({ &src, &dst }, data_layout); in compute_target()
359 …void setup(TensorShape input_shape, Size2D output_tile, DataLayout data_layout, DataType data_type) in setup() argument
365 … _target = compute_target(input_shape, output_shape, winograd_info, data_layout, data_type); in setup()
372 const DataLayout data_layout_src = src.info()->data_layout(); in mix_layout()
373 const DataLayout data_layout_dst = dst.info()->data_layout(); in mix_layout()
411 …nsorShape &output_shape, const WinogradInfo &winograd_info, DataLayout data_layout, DataType data_… in compute_target() argument
413 if(data_layout == DataLayout::NHWC) in compute_target()
419 …orType src = create_tensor<TensorType>(input_shape, data_type, 1, QuantizationInfo(), data_layout); in compute_target()
429 add_padding_x({ &src, &dst }, data_layout); in compute_target()
483 const DataLayout data_layout_src = src.info()->data_layout(); in mix_layout()
484 const DataLayout data_layout_dst = dst.info()->data_layout(); in mix_layout()