Home
last modified time | relevance | path

Searched refs:body_name (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/converters/
Dcontrol_flow_deprecated_py2.py43 def _create_cond_branch(self, body_name, aliased_orig_names, argument
82 body_name=body_name,
92 template, body_name=body_name, body=body, return_stmt=return_stmt)
94 def _create_cond_expr(self, results, test, body_name, orelse_name, argument
108 body_name=body_name,
122 body_name=body_name,
260 body_name = self.ctx.namer.new_symbol('if_true', body_scope.referenced)
298 body_name,
318 cond_expr = self._create_cond_expr(cond_results, cond_var_name, body_name,
445 body_name=self.ctx.namer.new_symbol('loop_body', reserved_symbols),
[all …]
Dcontrol_flow.py244 body_name=self.ctx.namer.new_symbol('if_body', reserved),
295 body_name=self.ctx.namer.new_symbol('loop_body', reserved),
380 body_name=self.ctx.namer.new_symbol('loop_body', reserved),
/external/tensorflow/tensorflow/core/common_runtime/
Dcontrol_flow_deps_to_chains.cc67 string body_name = attrs.at("body").func().name(); in Run() local
68 auto* body_graph = flib_def->Find(body_name); in Run()
310 flib_def->GetStackTraces(body_name)), in Run()
324 graph_lib->GetStackTraces(body_name)), in Run()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dregion_control_flow_to_functional.cc409 std::string cond_name, body_name; in ConvertWhileOp() local
413 body_name = tti.callee_names[1].str(); in ConvertWhileOp()
431 body_name = GetName(while_region, "_body"); in ConvertWhileOp()
432 ExtractSingleBlockRegion(while_region.body(), body_name, extern_values, in ConvertWhileOp()
445 while_region.getLoc(), new_result_types, new_inputs, cond_name, body_name, in ConvertWhileOp()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc420 NameAttrList body_name; in FunctionalizeLoop() local
421 body_name.set_name(library->UniqueFunctionName("_functionalize_body_")); in FunctionalizeLoop()
427 GraphToFunctionDef(*body_graph, body_name.name(), &body_fdef)); in FunctionalizeLoop()
437 builder.Attr("body", body_name); in FunctionalizeLoop()
Dfunctionalize_cond.cc756 NameAttrList body_name; in BuildIfNode() local
757 body_name.set_name(library->UniqueFunctionName( in BuildIfNode()
768 body_name.name(), &body_fdef)); in BuildIfNode()
770 builder.Attr(branch_name[branch_index], body_name); in BuildIfNode()
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc186 string body_name = in TEST_F() local
189 EXPECT_TRUE(TF_GraphOperationByName(graph_, body_name.c_str()) == nullptr); in TEST_F()
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py118 body_name = util.unique_fn_name(scope, "body")
223 body_name,
229body_name, collections=ops.get_default_graph()._collections), # pylint: disable=protected-access
Dfunctional_ops.py1039 body_name = "%s_Body" % forbody.name
1041 @function.Defun(*body_sig, func_name=body_name)
/external/autotest/client/cros/faft/utils/
Dflashrom_handler.py35 def __init__(self, sig_name, body_name, fwid_name=None): argument
45 self._body_name = body_name
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py862 body_name = node.attr["body"].func.name
865 if function_def.signature.name == body_name:
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py642 body_name = self._node.attr["body"].func.name
643 body = self._enclosing_graph.functions[body_name].converted_self().function
/external/tensorflow/tensorflow/core/ir/
Dops.cc1082 SymbolRefAttr body_name = op.body().getName(); in VerifyWhileLikeOp() local