Searched refs:body_def (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 1047 def _visit_if_stmt(self, cond, body_def, orelse_def, get_state, out_symbols, argument
1062 self.visit_block(body_def.body)
1130 def _visit_for_stmt(self, range_, body_def, get_state, loop_carried, node): argument
1153 assert len(body_def.args.args) == 1
1154 it_name = body_def.args.args[0].id
1171 self.visit_block(body_def.body)
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | control_flow_deprecated_py2.py | 301 body_def = self._create_cond_branch(
328 undefined_assigns + composite_defs + body_def + orelse_def +
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 676 const FunctionDef *body_def = fld.Find("oc_body_host_while_body_fn"); in TEST_F() local
677 EXPECT_NE(body_def, nullptr); in TEST_F()
679 for (const auto &node_def : body_def->node_def()) { in TEST_F()
|