Python ConstShapeOp Examples

Example #1

File: shape.py Project: puremourning/llvm-project

 def const_shape_tensor(arg):
     return shape.ConstShapeOp(
         DenseElementsAttr.get(np.array([10, 20])))

Example #2

 def const_shape_tensor(arg):
     return shape.ConstShapeOp(
         DenseElementsAttr.get(np.array([10, 20], dtype=np.int64),
                               type=IndexType.get()))

Example #3

 def const_shape_tensor(arg):
     return shape.ConstShapeOp(
         RankedTensorType.get((2, ), indexT),
         DenseElementsAttr.get(np.array([10, 20])))