Lines Matching refs:data_layout
49 …Info pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout, QuantizationIn… in setup() argument
51 …_target = compute_target(input_shape, data_type, data_layout, pool_info, rois_shape, qinfo, out… in setup()
63 …nst ROIPoolingLayerInfo &pool_info, TensorShape rois_shape, DataLayout data_layout = DataLayout::N…
76 …led_width = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
77 …led_height = static_cast<float>((shape[get_data_layout_dimension_index(data_layout, DataLayoutDime…
106 DataLayout data_layout, in compute_target() argument
115 … TensorType src = create_tensor<TensorType>(input_shape, data_type, 1, qinfo, data_layout); in compute_target()
120 …sorType dst = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout); in compute_target()
141 generate_rois(AccessorType(rois_tensor), input_shape, pool_info, rois_shape, data_layout); in compute_target()
179 DataLayout data_layout, QuantizationInfo qinfo, QuantizationInfo output_qinfo) in setup() argument
182 … data_type, data_layout, qinfo, output_qinfo); in setup()
191 … ROIPoolingLayerInfo pool_info, TensorShape rois_shape, DataType data_type, DataLayout data_layout) in setup() argument
193 …, AccessorType, FunctionType, T>::setup(input_shape, pool_info, rois_shape, data_type, data_layout, in setup()