Lines Matching defs:weightShape

769   ShapeAdaptor weightShape(adaptor.getWeight().getType());
780   if (weightShape.hasRank()) {
781     outShape[1] = weightShape.getDimSize(0);
1743   ShapeAdaptor weightShape(adaptor.getWeight().getType());
1744   if (weightShape.hasRank()) {
1745     outputShape[3] = weightShape.getDimSize(0);
1746     weightHeight = weightShape.getDimSize(1);
1747     weightWidth = weightShape.getDimSize(2);
1812   ShapeAdaptor weightShape(adaptor.getWeight().getType());
1813   if (weightShape.hasRank()) {
1814     outputShape[4] = weightShape.getDimSize(0);
1815     weightDepth = weightShape.getDimSize(1);
1816     weightHeight = weightShape.getDimSize(2);
1817     weightWidth = weightShape.getDimSize(3);
1908   ShapeAdaptor weightShape(adaptor.getWeight().getType());
1909   if (weightShape.hasRank()) {
1910     weightHeight = weightShape.getDimSize(0);
1911     weightWidth = weightShape.getDimSize(1);
1913                         ? weightShape.getDimSize(2)
1915     depthChannels = weightShape.getDimSize(3);
1987   ShapeAdaptor weightShape(adaptor.getFilter().getType());
1988   if (weightShape.hasRank()) {
1990                          ? weightShape.getDimSize(0)
1992     weightHeight = weightShape.getDimSize(1);
1993     weightWidth = weightShape.getDimSize(2);