Lines Matching full:inputs
17 [[1,3,224,224]], # set the first shape for all inputs
18 [[1,3,192,192]], # set the second shape for all inputs
19 [[1,3,160,160]], # set the third shape for all inputs
114 rknn_input inputs[io_num.n_input];
115 memset(inputs, 0, io_num.n_input * sizeof(rknn_input));
121 inputs[i].index = i;
122 inputs[i].pass_through = 0;
123 inputs[i].type = RKNN_TENSOR_UINT8;
124 inputs[i].fmt = RKNN_TENSOR_NHWC;
125 inputs[i].buf = imgs[i].data;
126 inputs[i].size = imgs[i].total() * imgs[i].channels();
130 ret = rknn_inputs_set(ctx, io_num.n_input, inputs);