Esempi in Python per ConstShapeOp

Esempio n. 1

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

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

Esempio n. 2

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

Esempio n. 3

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