Exemplos de ConstShapeOp em Python

Exemplo n.º 1

Arquivo: shape.py Projeto: puremourning/llvm-project

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

Exemplo n.º 2

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

Exemplo n.º 3

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