mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
02c48cced9
Summary: Stacked on top of https://github.com/pytorch/pytorch/pull/17386 Brings us to 1014/1106 of writing. Pull Request resolved: https://github.com/pytorch/pytorch/pull/17385 Differential Revision: D14248008 Pulled By: cpuhrsch fbshipit-source-id: 033e00de91e3edf7ae01ca03ebe436c0446b3b5c
30 lines
1.1 KiB
Plaintext
30 lines
1.1 KiB
Plaintext
graph(%cond : Tensor):
|
|
%1 : int[] = prim::Constant[value=[1]]()
|
|
%2 : int[] = prim::Constant[value=[0]]()
|
|
%3 : int = prim::Constant[value=3]()
|
|
%4 : Double(2) = prim::Constant[value= 4 4 [ Variable[CPUDoubleType]{2} ]]()
|
|
%5 : Double(2) = prim::Constant[value= 1 1 [ Variable[CPUDoubleType]{2} ]]()
|
|
%c.1 : int = prim::Constant[value=0]()
|
|
%a : int = prim::Constant[value=1]()
|
|
%d : string = prim::Constant[value="abc"]()
|
|
%e : string = prim::Constant[value="bcd"]()
|
|
%10 : int? = prim::Constant()
|
|
%11 : Device? = prim::Constant()
|
|
%12 : bool = prim::Bool(%cond)
|
|
%c : int, %y : Tensor = prim::If(%12)
|
|
block0():
|
|
-> (%3, %4)
|
|
block1():
|
|
%y.2 : Tensor = aten::rand(%2, %10, %10, %11)
|
|
%16 : bool = prim::Bool(%cond)
|
|
%y.4 : Tensor = prim::If(%16)
|
|
block0():
|
|
%y.3 : Tensor = aten::rand(%1, %10, %10, %11)
|
|
-> (%y.3)
|
|
block1():
|
|
-> (%y.2)
|
|
= prim::Print(%d, %e, %d, %5, %y.4, %5)
|
|
-> (%c.1, %y.4)
|
|
%19 : (int, int, int, Tensor, Tensor) = prim::TupleConstruct(%a, %3, %c, %5, %y)
|
|
return (%19)
|