Searched refs:im2col_array (Results 1 – 2 of 2) sorted by relevance
33 auto& im2col_array = model->GetArray(op->outputs[1]); in HardcodeMinMaxForIm2colArray() local34 if (im2col_array.minmax) { in HardcodeMinMaxForIm2colArray()42 CHECK(!im2col_array.minmax); in HardcodeMinMaxForIm2colArray()43 auto& im2col_minmax = im2col_array.GetOrCreateMinMax(); in HardcodeMinMaxForIm2colArray()
158 auto& im2col_array = model->GetArray(op->outputs[1]); in ProcessConvOperator() local159 im2col_array.copy_shape(Shape{output_shape.dims(0), output_shape.dims(1), in ProcessConvOperator()246 auto& im2col_array = model->GetArray(op->outputs[1]); in ProcessTransposeConvOperator() local247 im2col_array.copy_shape( in ProcessTransposeConvOperator()