Lines Matching refs:data_layout
50 …id setup(TensorShape shape, PoolingLayerInfo pool_info, DataType data_type, DataLayout data_layout) in setup() argument
60 … _target = compute_target(shape, pool_info, data_type, data_layout, input_qinfo, output_qinfo); in setup()
85 DataType data_type, DataLayout data_layout, in compute_target() argument
89 if(data_layout == DataLayout::NHWC) in compute_target()
95 … src = create_tensor<TensorType>(input_shape, data_type, 1, input_qinfo, data_layout); in compute_target()
97 …e dst = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
98 … unpooled = create_tensor<TensorType>(input_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
99 … indices = create_tensor<TensorType>(dst_shape, DataType::U32, 1, output_qinfo, data_layout); in compute_target()
154 …_type, Size2D pool_size, PadStrideInfo pad_stride_info, DataType data_type, DataLayout data_layout) in setup() argument
156 …rType, F1, F2, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_inf… in setup()
157 … data_type, data_layout); in setup()