Searched defs:inputShape (Results 1 – 2 of 2) sorted by relevance
78 const Shape inputShape = model->getShape(ins[0]); in average_pool_2d() local149 const Shape inputShape = model->getShape(ins[0]); in conv_2d() local199 const Shape inputShape = model->getShape(ins[0]); in depthwise_conv_2d() local266 const Shape inputShape = model->getShape(ins[0]); in l2_pool_2d() local350 const Shape inputShape = model->getShape(ins[0]); in max_pool_2d() local533 const Shape inputShape = model->getShape(ins[0]); in average_pool_2d() local608 const Shape inputShape = model->getShape(ins[0]); in conv_2d() local664 const Shape inputShape = model->getShape(ins[0]); in depthwise_conv_2d() local781 const Shape inputShape = model->getShape(ins[0]); in max_pool_2d() local
158 const Shape inputShape = model->getShape(ins[0]); in conv_2d() local218 const Shape inputShape = model->getShape(ins[0]); in depthwise_conv_2d() local279 const Shape inputShape = model->getShape(ins[0]); in dequantize() local295 const Shape inputShape = model->getShape(ins[0]); in fully_connected() local