Home
last modified time | relevance | path

Searched defs:cond_func (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc70 AttrValue cond_func; in TEST() local
192 AttrValue cond_func; in TEST() local
297 AttrValue cond_func; in TEST() local
421 AttrValue cond_func; in TEST() local
535 AttrValue cond_func; in TEST() local
Dlower_functional_ops_test.cc158 AttrValue cond_func; in TEST() local
208 NameAttrList cond_func; in IfWithWhileThen() local
Dexecutor_test.cc671 AttrValue cond_func; in BM_WhileLoopHelper() local
/external/tensorflow/tensorflow/core/kernels/
Dwhile_op_test.cc189 AttrValue cond_func; in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils_test.cc843 const FunctionDef* cond_func = cond_lib.Find("__inference_test_function_19"); in TEST() local
Dfunction_utils.cc220 const FunctionDef* cond_func = in IsNodeStateful() local
/external/pytorch/torch/_higher_order_ops/
Dcond.py449 def cond_func(ctx, pred, true_fn, false_fn, inputs): function
/external/tensorflow/tensorflow/core/transforms/functional_to_region/
Dimpl.cc432 GraphFuncOp cond_func = this->LookupFunc(op.cond()); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/data/
Ddataset_utils.cc373 const FunctionDef* cond_func = in IsNodeStateful() local
/external/tensorflow/tensorflow/core/transforms/region_to_functional/
Dimpl.cc962 GraphFuncOp cond_func = in matchAndRewrite() local
/external/tensorflow/tensorflow/core/ir/
Dops.cc1084 auto cond_func = symbol_table.lookupNearestSymbolFrom<GraphFuncOp>( in VerifyWhileLikeOp() local
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc661 NameAttrList cond_func; in PostprocessLiftedArgsForWhile() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1734 NameAttrList cond_func; in FindArgsToLiftForWhileNode() local