torch-mlir/test/npcomp-run-mlir/pad.mlir

19 lines
714 B
MLIR
Raw Normal View History

2021-01-28 02:55:26 +08:00
// RUN: npcomp-run-mlir %s \
// RUN: -invoke pad \
// RUN: -arg-value="dense<[1.2, 3.4]> : tensor<2xf32>" \
// RUN: -shared-libs=%npcomp_runtime_shlib 2>&1 \
// RUN: | FileCheck %s
// CHECK: output #0: dense<
// CHECK-SAME: [0.000000e+00, 1.200000e+00, 3.400000e+00, 0.000000e+00, 0.000000e+00]
// CHECK-SAME: > : tensor<5xf32>
func @pad(%arg0: tensor<?xf32> ) -> tensor<?xf32> {
%lowerExpansion = shape.const_shape [1] : tensor<?xindex>
%upperExpansion = shape.const_shape [2] : tensor<?xindex>
%fillVal = constant 0.0 : f32
%0 = tcp.pad %arg0, %lowerExpansion, %upperExpansion, %fillVal : (tensor<?xf32>, tensor<?xindex>, tensor<?xindex>, f32) -> tensor<?xf32>
return %0 : tensor<?xf32>
}