Lines Matching refs:weightShape

429   ShapeAdaptor weightShape = operands.getShape(1);  in inferReturnTypeComponents()  local
440 if (weightShape.hasRank()) { in inferReturnTypeComponents()
441 outShape[1] = weightShape.getDimSize(0); in inferReturnTypeComponents()
1035 ShapeAdaptor weightShape = operands.getShape(adaptor.getWeight()); in inferReturnTypeComponents() local
1036 if (weightShape.hasRank()) { in inferReturnTypeComponents()
1037 outputShape[3] = weightShape.getDimSize(0); in inferReturnTypeComponents()
1038 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()
1039 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()
1105 ShapeAdaptor weightShape = operands.getShape(adaptor.getWeight()); in inferReturnTypeComponents() local
1106 if (weightShape.hasRank()) { in inferReturnTypeComponents()
1107 outputShape[4] = weightShape.getDimSize(0); in inferReturnTypeComponents()
1108 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()
1109 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()
1110 weightDepth = weightShape.getDimSize(3); in inferReturnTypeComponents()
1197 ShapeAdaptor weightShape = operands.getShape(adaptor.getWeight()); in inferReturnTypeComponents() local
1198 if (weightShape.hasRank()) { in inferReturnTypeComponents()
1199 weightHeight = weightShape.getDimSize(0); in inferReturnTypeComponents()
1200 weightWidth = weightShape.getDimSize(1); in inferReturnTypeComponents()
1202 ? weightShape.getDimSize(2) in inferReturnTypeComponents()
1204 depthChannels = weightShape.getDimSize(3); in inferReturnTypeComponents()
1276 ShapeAdaptor weightShape = operands.getShape(adaptor.getFilter()); in inferReturnTypeComponents() local
1277 if (weightShape.hasRank()) { in inferReturnTypeComponents()
1279 ? weightShape.getDimSize(0) in inferReturnTypeComponents()
1281 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()
1282 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()