Searched refs:while_compute (Results 1 – 1 of 1) sorted by relevance
102 def while_compute(x): function108 result = xla.compile(while_compute, inputs=[array_ops.ones((10, 5, 3))])