Lines Matching refs:winograd_info
211 WinogradInfo winograd_info(output_tile, in compute_reference()
218 …pe = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_type), winograd_info); in compute_reference()
219 … = compute_winograd_filter_transform_shape(TensorInfo(weights_shape, 1, data_type), winograd_info); in compute_reference()
222 …mpute_winograd_output_transform_shape(TensorInfo(batched_gemm_shape, 1, data_type), winograd_info); in compute_reference()
228 …nsform_out = reference::winograd_input_transform<T1>(src_t1, input_transform_shape, winograd_info); in compute_reference()
230 …_out = reference::winograd_filter_transform<T1>(weights_t1, filter_transform_shape, winograd_info); in compute_reference()
232 …rence::winograd_output_transform<T1>(batched_gemm, bias_t1, output_transform_shape, winograd_info); in compute_reference()
247 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d… in setup() argument
249 …ape = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_type), winograd_info); in setup()
251 …_target = compute_target(input_shape, output_shape, winograd_info, data_layout, d… in setup()
252 … _reference = compute_reference(input_shape, output_shape, winograd_info, data_type); in setup()
297 …ape input_shape, const TensorShape &output_shape, const WinogradInfo &winograd_info, DataLayout da… in compute_target() argument
309 transf.configure(&src, &dst, winograd_info); in compute_target()
338 …pe &input_shape, const TensorShape &output_shape, const WinogradInfo &winograd_info, DataType data… in compute_reference() argument
346 return reference::winograd_input_transform<T>(src, output_shape, winograd_info); in compute_reference()
361 …WinogradInfo winograd_info(output_tile, Size2D(input_shape[0], input_shape[1]), Size2D() /* Not ne… in setup()
362 …pe = compute_winograd_filter_transform_shape(TensorInfo(input_shape, 1, data_type), winograd_info); in setup()
365 … _target = compute_target(input_shape, output_shape, winograd_info, data_layout, data_type); in setup()
366 _reference = compute_reference(input_shape, output_shape, winograd_info, data_type); in setup()
411 …ape input_shape, const TensorShape &output_shape, const WinogradInfo &winograd_info, DataLayout da… in compute_target() argument
424 filter_transform.configure(&src, &dst, winograd_info); in compute_target()
453 …pe &input_shape, const TensorShape &output_shape, const WinogradInfo &winograd_info, DataType data… in compute_reference() argument
461 return reference::winograd_filter_transform<T>(src, output_shape, winograd_info); in compute_reference()
474 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataType data_type, ActivationLaye…
476 _target = compute_target(input_shape, winograd_info, data_type, act_info);
477 _reference = compute_reference(input_shape, winograd_info, data_type, act_info);
522 …TensorType compute_target(const TensorShape &input_shape, const WinogradInfo &winograd_info, DataT… in compute_target() argument
524 …pe = compute_winograd_output_transform_shape(TensorInfo(input_shape, 1, data_type), winograd_info); in compute_target()
528 …reate_tensor<TensorType>(output_shape[get_data_layout_dimension_index(winograd_info.output_data_la… in compute_target()
529 …te_tensor<TensorType>(output_shape, data_type, 1, QuantizationInfo(), winograd_info.output_data_la… in compute_target()
533 output_transform.configure(&src, &bias, &dst, winograd_info, act_info); in compute_target()
539 add_padding_x({ &src, &bias, &dst }, winograd_info.output_data_layout); in compute_target()
566 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, WinogradInfo winograd_info, Data… in compute_reference() argument
568 winograd_info.output_data_layout = DataLayout::NCHW; in compute_reference()
569 … = compute_winograd_output_transform_shape(TensorInfo(input_shape, 1, data_type), winograd_info); in compute_reference()
579 … winograd_output = reference::winograd_output_transform<T>(src, bias, output_shape, winograd_info); in compute_reference()