Searched refs:im2col_id (Results 1 – 2 of 2) sorted by relevance
52 int im2col_id = kTensorNotAllocated; member94 if (data->im2col_id == kTensorNotAllocated) { in AllocateIm2colTensorIfRequired()95 context->AddTensors(context, 1, &data->im2col_id); in AllocateIm2colTensorIfRequired()96 context->tensors[data->im2col_id].type = kTfLiteFloat32; in AllocateIm2colTensorIfRequired()101 node->temporaries->data[data->im2col_index] = data->im2col_id; in AllocateIm2colTensorIfRequired()
62 int im2col_id = kTensorNotAllocated; member177 if (data->im2col_id == kTensorNotAllocated) { in AllocateTemporaryTensorsIfRequired()178 context->AddTensors(context, 1, &data->im2col_id); in AllocateTemporaryTensorsIfRequired()337 node->temporaries->data[data->im2col_index] = data->im2col_id; in Prepare()