Lines Matching refs:data_layout
44 DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in setup() argument
46 …put_shape, block_shape_shape, paddings_shape, output_shape, data_type, data_layout, quantization_i… in setup()
64 … DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in compute_target() argument
66 if(data_layout == DataLayout::NHWC) in compute_target()
73 …input = create_tensor<TensorType>(input_shape, data_type, 1, quantization_info, data_layout); in compute_target()
76 …utput = create_tensor<TensorType>(output_shape, data_type, 1, quantization_info, data_layout); in compute_target()
103 …const int idx_width = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WID… in compute_target()
145 DataType data_type, DataLayout data_layout) in setup() argument
147 …put_shape, block_shape_shape, paddings_shape, output_shape, data_type, data_layout, QuantizationIn… in setup()
157 DataType data_type, DataLayout data_layout, QuantizationInfo quantization_info) in setup() argument
159 …put_shape, block_shape_shape, paddings_shape, output_shape, data_type, data_layout, quantization_i… in setup()