Lines Matching refs:data_layout
49 …Info pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout, QuantizationIn… in setup() argument
52 …_target = compute_target(input_shape, data_type, data_layout, pool_info, rois_shape, qinfo… in setup()
64 …nst ROIPoolingLayerInfo &pool_info, TensorShape rois_shape, DataLayout data_layout = DataLayout::N…
77 …led_width = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
78 …led_height = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
117 DataLayout data_layout, in compute_target() argument
123 if(data_layout == DataLayout::NHWC) in compute_target()
131 … TensorType src = create_tensor<TensorType>(input_shape, data_type, 1, qinfo, data_layout); in compute_target()
135 …e dst = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
156 generate_rois(AccessorType(rois_tensor), input_shape, pool_info, rois_shape, data_layout); in compute_target()
193 … ROIPoolingLayerInfo pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout) in setup() argument
195 …sorType, FunctionType, T, TRois>::setup(input_shape, pool_info, rois_shape, data_type, data_layout, in setup()
206 DataLayout data_layout, QuantizationInfo qinfo, QuantizationInfo output_qinfo) in setup() argument
209 … data_type, data_layout, qinfo, output_qinfo); in setup()