Home
last modified time | relevance | path

Searched defs:body_func (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc72 AttrValue body_func; in TEST() local
194 AttrValue body_func; in TEST() local
299 AttrValue body_func; in TEST() local
423 AttrValue body_func; in TEST() local
537 AttrValue body_func; in TEST() local
Dlower_functional_ops_test.cc160 AttrValue body_func; in TEST() local
210 NameAttrList body_func; in IfWithWhileThen() local
Dexecutor_test.cc673 AttrValue body_func; in BM_WhileLoopHelper() local
/external/tensorflow/tensorflow/python/framework/
Dpy_context_manager_pybind.cc27 py::handle body_func) { in TestPyContextManager()
/external/tensorflow/tensorflow/core/transforms/functional_to_region/
Dimpl.cc433 GraphFuncOp body_func = this->LookupFunc(op.body()); in matchAndRewrite() local
475 GraphFuncOp body_func = LookupFunc(op.body()); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/kernels/
Dwhile_op_test.cc192 AttrValue body_func; in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils.cc222 const FunctionDef* body_func = in IsNodeStateful() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc318 const FunctionDef* body_func = lookup_fld->Find(name); in FindOrInsert() local
414 const FunctionDef* body_func = lookup_fld->Find(fn_name); in PropagateConstIntoAndAroundWhileNode() local
/external/tensorflow/tensorflow/core/ir/
Dops.cc1091 auto body_func = symbol_table.lookupNearestSymbolFrom<GraphFuncOp>( in VerifyWhileLikeOp() local
1109 auto body_func = in verifySymbolUses() local
/external/tensorflow/tensorflow/core/data/
Ddataset_utils.cc376 const FunctionDef* body_func = in IsNodeStateful() local
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc586 NameAttrList body_func; in PostprocessLiftedArgsForWhile() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1754 NameAttrList body_func; in FindArgsToLiftForWhileNode() local