Untitled Diff

Created Diff never expires
0 removals
51 lines
24 additions
51 lines
#[version = "0.0.5"]
#[version = "0.0.5"]
def @main(%data: Tensor[(1, 3, 128, 128), float32], hash="8e8c38e8c06d4b9e") -> Tensor[(1, 32, 126, 126), float32] {
def @main(%data: Tensor[(1, 3, 128, 128), float32], hash="8e8c38e8c06d4b9e") -> Tensor[(1, 32, 126, 126), float32] {
%0 = fn (%p02: Tensor[(1, 3, 128, 128), float32], %p12: Tensor[(32, 3, 5, 5), float32], hash="cb1af1d5e8b15e79", out_layout="", kernel_layout="OIHW", Primitive=1, data_layout="NCHW") -> Tensor[(1, 32, 126, 126), float32] {
%0 = fn (%p02: Tensor[(1, 3, 128, 128), float32], %p12: Tensor[(32, 3, 5, 5), float32], hash="cb1af1d5e8b15e79", out_layout="", kernel_layout="OIHW", Primitive=1, data_layout="NCHW") -> Tensor[(1, 32, 126, 126), float32] {
nn.conv2d(%p02, %p12, padding=[1, 1, 1, 1], channels=32, kernel_size=[5, 5]) /* ty=Tensor[(1, 32, 126, 126), float32] */
nn.conv2d(%p02, %p12, padding=[1, 1, 1, 1], channels=32, kernel_size=[5, 5]) /* GeneratedSource */ /* ty=Tensor[(1, 32, 126, 126), float32] */
};
};
%1 = fn (%p07: Tensor[(32), float32], %p15: float32, Primitive=1, hash="85ca91fe9057771b") -> Tensor[(32), float32] {
%1 = fn (%p07: Tensor[(32), float32], %p15: float32, Primitive=1, hash="85ca91fe9057771b") -> Tensor[(32), float32] {
add(%p07, %p15) /* ty=Tensor[(32), float32] */
add(%p07, %p15) /* GeneratedSource */ /* ty=Tensor[(32), float32] */
};
};
%2 = %1(meta[relay.Constant][1] /* ty=Tensor[(32), float32] */, 1e-05f /* ty=float32 */) /* ty=Tensor[(32), float32] */;
%2 = %1(meta[relay.Constant][1] /* ty=Tensor[(32), float32] */, 1e-05f /* ty=float32 */) /* GeneratedSource */ /* ty=Tensor[(32), float32] */;
%3 = fn (%p06: Tensor[(32), float32], Primitive=1, hash="9056ed434d7a90e4") -> Tensor[(32), float32] {
%3 = fn (%p06: Tensor[(32), float32], Primitive=1, hash="9056ed434d7a90e4") -> Tensor[(32), float32] {
sqrt(%p06) /* ty=Tensor[(32), float32] */
sqrt(%p06) /* GeneratedSource */ /* ty=Tensor[(32), float32] */
};
};
%4 = %3(%2) /* ty=Tensor[(32), float32] */;
%4 = %3(%2) /* GeneratedSource */ /* ty=Tensor[(32), float32] */;
%5 = fn (%p05: float32, %p14: Tensor[(32), float32], Primitive=1, hash="a594e3a2d400dd46") -> Tensor[(32), float32] {
%5 = fn (%p05: float32, %p14: Tensor[(32), float32], Primitive=1, hash="a594e3a2d400dd46") -> Tensor[(32), float32] {
divide(%p05, %p14) /* ty=Tensor[(32), float32] */
divide(%p05, %p14) /* GeneratedSource */ /* ty=Tensor[(32), float32] */
};
};
%6 = %5(1f /* ty=float32 */, %4) /* ty=Tensor[(32), float32] */;
%6 = %5(1f /* ty=float32 */, %4) /* GeneratedSource */ /* ty=Tensor[(32), float32] */;
%7 = fn (%p04: Tensor[(32), float32], %p13: Tensor[(32), float32], Primitive=1, hash="c3a46c9c4e5b4441") -> Tensor[(32), float32] {
%7 = fn (%p04: Tensor[(32), float32], %p13: Tensor[(32), float32], Primitive=1, hash="c3a46c9c4e5b4441") -> Tensor[(32), float32] {
multiply(%p04, %p13) /* ty=Tensor[(32), float32] */
multiply(%p04, %p13) /* GeneratedSource */ /* ty=Tensor[(32), float32] */
};
};
%8 = %7(%6, meta[relay.Constant][2] /* ty=Tensor[(32), float32] */) /* ty=Tensor[(32), float32] */;
%8 = %7(%6, meta[relay.Constant][2] /* ty=Tensor[(32), float32] */) /* GeneratedSource */ /* ty=Tensor[(32), float32] */;
%9 = fn (%p03: Tensor[(32), float32], Primitive=1, relay.reshape_only=1, hash="2bf6b07e7d6f8c28") -> Tensor[(32, 1, 1), float32] {
%9 = fn (%p03: Tensor[(32), float32], Primitive=1, relay.reshape_only=1, hash="2bf6b07e7d6f8c28") -> Tensor[(32, 1, 1), float32] {
expand_dims(%p03, axis=1, num_newaxis=2) /* ty=Tensor[(32, 1, 1), float32] */
expand_dims(%p03, axis=1, num_newaxis=2) /* GeneratedSource */ /* ty=Tensor[(32, 1, 1), float32] */
};
};
%10 = %0(%data, meta[relay.Constant][0] /* ty=Tensor[(32, 3, 5, 5), float32] */) /* ty=Tensor[(1, 32, 126, 126), float32] */;
%10 = %0(%data, meta[relay.Constant][0] /* ty=Tensor[(32, 3, 5, 5), float32] */) /* GeneratedSource */ /* ty=Tensor[(1, 32, 126, 126), float32] */;
%11 = %9(%8) /* ty=Tensor[(32, 1, 1), float32] */;
%11 = %9(%8) /* GeneratedSource */ /* ty=Tensor[(32, 1, 1), float32] */;
%12 = fn (%p01: Tensor[(1, 32, 126, 126), float32], %p11: Tensor[(32, 1, 1), float32], Primitive=1, hash="1d1c67b1a2f16e88") -> Tensor[(1, 32, 126, 126), float32] {
%12 = fn (%p01: Tensor[(1, 32, 126, 126), float32], %p11: Tensor[(32, 1, 1), float32], Primitive=1, hash="1d1c67b1a2f16e88") -> Tensor[(1, 32, 126, 126), float32] {
multiply(%p01, %p11) /* ty=Tensor[(1, 32, 126, 126), float32] */
multiply(%p01, %p11) /* GeneratedSource */ /* ty=Tensor[(1, 32, 126, 126), float32] */
};
};
%13 = fn (%p011: Tensor[(32), float32], Primitive=1, hash="d071045a392451ec") -> Tensor[(32), float32] {
%13 = fn (%p011: Tensor[(32), float32], Primitive=1, hash="d071045a392451ec") -> Tensor[(32), float32] {
negative(%p011) /* ty=Tensor[(32), float32] */
negative(%p011) /* GeneratedSource */ /* ty=Tensor[(32), float32] */
};
};
%14 = %13(meta[relay.Constant][3] /* ty=Tensor[(32), float32] */) /* ty=Tensor[(32), float32] */;
%14 = %13(meta[relay.Constant][3] /* ty=Tensor[(32), float32] */) /* GeneratedSource */ /* ty=Tensor[(32), float32] */;
%15 = fn (%p010: Tensor[(32), float32], %p17: Tensor[(32), float32], Primitive=1, hash="c3a46c9c4e5b4441") -> Tensor[(32), float32] {
%15 = fn (%p010: Tensor[(32), float32], %p17: Tensor[(32), float32], Primitive=1, hash="c3a46c9c4e5b4441") -> Tensor[(32), float32] {
multiply(%p010, %p17) /* ty=Tensor[(32), float32] */
multiply(%p010, %p17) /* GeneratedSource */ /* ty=Tensor[(32), float32] */
};
};
%16 = %15(%14, %8) /* ty=Tensor[(32), float32] */;
%16 = %15(%14, %8) /* GeneratedSource */ /* ty=Tensor[(32), float32] */;
%17 = fn (%p09: Tensor[(32), float32], %p16: Tensor[(32), float32], Primitive=1, hash="8d477b3724fd7eaa") -> Tensor[(32), float32] {
%17 = fn (%p09: Tensor[(32), float32], %p16: Tensor[(32), float32], Primitive=1, hash="8d477b3724fd7eaa") -> Tensor[(32), float32] {
add(%p09, %p16) /* ty=Tensor[(32), float32] */
add(%p09, %p16) /* GeneratedSource */ /* ty=Tensor[(32), float32] */
};
};
%18 = %17(%16, meta[relay.Constant][4] /* ty=Tensor[(32), float32] */) /* ty=Tensor[(32), float32] */;
%18 = %17(%16, meta[relay.Constant][4] /* ty=Tensor[(32), float32] */) /* GeneratedSource */ /* ty=Tensor[(32), float32] */;
%19 = fn (%p08: Tensor[(32), float32], Primitive=1, relay.reshape_only=1, hash="2bf6b07e7d6f8c28") -> Tensor[(32, 1, 1), float32] {
%19 = fn (%p08: Tensor[(32), float32], Primitive=1, relay.reshape_only=1, hash="2bf6b07e7d6f8c28") -> Tensor[(32, 1, 1), float32] {
expand_dims(%p08, axis=1, num_newaxis=2) /* ty=Tensor[(32, 1, 1), float32] */
expand_dims(%p08, axis=1, num_newaxis=2) /* GeneratedSource */ /* ty=Tensor[(32, 1, 1), float32] */
};
};
%20 = %12(%10, %11) /* ty=Tensor[(1, 32, 126, 126), float32] */;
%20 = %12(%10, %11) /* GeneratedSource */ /* ty=Tensor[(1, 32, 126, 126), float32] */;
%21 = %19(%18) /* ty=Tensor[(32, 1, 1), float32] */;
%21 = %19(%18) /* GeneratedSource */ /* ty=Tensor[(32, 1, 1), float32] */;
%22 = fn (%p0: Tensor[(1, 32, 126, 126), float32], %p1: Tensor[(32, 1, 1), float32], Primitive=1, hash="b05990937918479d") -> Tensor[(1, 32, 126, 126), float32] {
%22 = fn (%p0: Tensor[(1, 32, 126, 126), float32], %p1: Tensor[(32, 1, 1), float32], Primitive=1, hash="b05990937918479d") -> Tensor[(1, 32, 126, 126), float32] {
add(%p0, %p1) /* ty=Tensor[(1, 32, 126, 126), float32] */
add(%p0, %p1) /* GeneratedSource */ /* ty=Tensor[(1, 32, 126, 126), float32] */
};
};
%22(%20, %21) /* ty=Tensor[(1, 32, 126, 126), float32] */
%22(%20, %21) /* GeneratedSource */ /* ty=Tensor[(1, 32, 126, 126), float32] */
}
}