Searched refs:getShape (Results 1 – 4 of 4) sorted by relevance
40 const Shape in1Shape = model->getShape(ins[0]); in addMul()41 const Shape in2Shape = model->getShape(ins[1]); in addMul()42 Shape outShape = model->getShape(outs[0]); in addMul()63 const Shape inShape = model->getShape(ins[0]); in pool()104 Shape outShape = model->getShape(outs[0]); in pool()142 inShapes[i] = model->getShape(ins[i]); in concatenation()144 Shape outShape = model->getShape(outs[0]); in concatenation()158 const Shape inputShape = model->getShape(ins[0]); in conv_2d()159 const Shape filterShape = model->getShape(ins[1]); in conv_2d()160 const Shape biasShape = model->getShape(ins[2]); in conv_2d()[all …]
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()[all …]
85 Shape getShape(uint32_t operand);
108 Shape Model::getShape(uint32_t operand) { in getShape() function in android::hardware::neuralnetworks::V1_0::implementation::hexagon::Model205 if (getShape(operand).type == OperandType::TENSOR_FLOAT32) { in createConvFilterTensor()238 if (getShape(operand).type == OperandType::TENSOR_FLOAT32) { in createFullyConnectedWeightTensor()