Searched refs:im2col_size (Results 1 – 1 of 1) sorted by relevance
339 TfLiteIntArray* im2col_size = TfLiteIntArrayCreate(4); in Prepare() local342 im2col_size->data[0] = output_size->data[0]; in Prepare()343 im2col_size->data[1] = output_size->data[1]; in Prepare()344 im2col_size->data[2] = output_size->data[2]; in Prepare()345 im2col_size->data[3] = input_depth * filter_height * filter_width; in Prepare()354 auto im2col_status = context->ResizeTensor(context, im2col, im2col_size); in Prepare()