Lines Matching refs:getShape
78 const Shape inputShape = model->getShape(ins[0]); in average_pool_2d()
115 const int32_t dims = model->getShape(ins[0]).dimensions.size(); in concatenation()
149 const Shape inputShape = model->getShape(ins[0]); in conv_2d()
150 const Shape filterShape = model->getShape(ins[1]); in conv_2d()
179 const Shape filterShape = model->getShape(ins[1]); in depthwise_conv_2d()
199 const Shape inputShape = model->getShape(ins[0]); in depthwise_conv_2d()
200 const Shape filterShape = model->getShape(ins[1]); in depthwise_conv_2d()
266 const Shape inputShape = model->getShape(ins[0]); in l2_pool_2d()
350 const Shape inputShape = model->getShape(ins[0]); in max_pool_2d()
533 const Shape inputShape = model->getShape(ins[0]); in average_pool_2d()
574 const int32_t dims = model->getShape(ins[0]).dimensions.size(); in concatenation()
608 const Shape inputShape = model->getShape(ins[0]); in conv_2d()
609 const Shape filterShape = model->getShape(ins[1]); in conv_2d()
664 const Shape inputShape = model->getShape(ins[0]); in depthwise_conv_2d()
665 const Shape filterShape = model->getShape(ins[1]); in depthwise_conv_2d()
781 const Shape inputShape = model->getShape(ins[0]); in max_pool_2d()