Searched refs:closure_elements (Results 1 – 4 of 4) sorted by relevance
84 set(stepper.closure_elements()))433 closure_elements = stepper.closure_elements()434 self.assertIn("x/read:0", closure_elements)435 self.assertIn("e:0", closure_elements)436 self.assertIn("f:0", closure_elements)514 set(stepper.closure_elements()))1076 closure_elements = stepper.closure_elements()1078 for element_name in closure_elements:1079 if element_name + ":0" in closure_elements:
251 closure_elements = [elem.name for elem in elem_list]298 closure_elements.append(inp.name)330 return sorted_nodes, closure_elements, ref_tensor_names345 def closure_elements(self): member in NodeStepper
344 closure_elements = node_stepper.closure_elements()350 if (node + ":0" not in closure_elements and351 node + ":1" not in closure_elements):
202 self._closure_elements = self._node_stepper.closure_elements()