Skip to content

Commit 8d9fb12

Browse files
cesarsouzamigueldeicaza
authored andcommitted
migueldeicazaGH-120: Adding support for tf.Cond (migueldeicaza#176)
1 parent 6c53399 commit 8d9fb12

6 files changed

Lines changed: 2337 additions & 1415 deletions

File tree

OpGenerator/OpGenerator.cs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -317,6 +317,10 @@ void Generate (OpDef oper)
317317
p ($"desc.AddInput ({ParamMap (arg.name)});");
318318
}
319319

320+
pi ("foreach ( TFOperation control in CurrentDependencies )");
321+
p ("desc.AddControlInput (control);");
322+
pd ("");
323+
320324
// If we have attributes
321325
if (required_attrs.Count > 0 || optional_attrs.Count > 0) {
322326
foreach (var attr in required_attrs) {

0 commit comments

Comments
 (0)