| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 70 AttrValue cond_func; in TEST() local
71 cond_func.mutable_func()->set_name("LessThanOrEqualToN"); in TEST()
78 .Attr("cond", cond_func) in TEST()
192 AttrValue cond_func; in TEST() local
193 cond_func.mutable_func()->set_name("LessThanOrEqualToN"); in TEST()
200 .Attr("cond", cond_func) in TEST()
297 AttrValue cond_func; in TEST() local
298 cond_func.mutable_func()->set_name("LessThanOrEqualToN"); in TEST()
307 .Attr("cond", cond_func) in TEST()
421 AttrValue cond_func; in TEST() local
[all …]
|
| D | lower_functional_ops_test.cc | 158 AttrValue cond_func; in TEST() local
159 cond_func.mutable_func()->set_name("WhileWithIfCond"); in TEST()
165 .Attr("cond", cond_func) in TEST()
208 NameAttrList cond_func; in IfWithWhileThen() local
209 cond_func.set_name("LessThanOrEqualToN"); in IfWithWhileThen()
228 {{"cond", cond_func}, in IfWithWhileThen()
|
| D | executor_test.cc | 622 AttrValue cond_func; in BM_WhileLoopHelper() local
623 cond_func.mutable_func()->set_name("LessThanOrEqualToN"); in BM_WhileLoopHelper()
630 .Attr("cond", cond_func) in BM_WhileLoopHelper()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils.cc | 220 const FunctionDef* cond_func = in IsNodeStateful() local
224 if ((cond_func != nullptr && in IsNodeStateful()
225 !IsFunctionStateful(library, *cond_func, skip_assert)) && in IsNodeStateful()
|
| D | function_utils_test.cc | 843 const FunctionDef* cond_func = cond_lib.Find("__inference_test_function_19"); in TEST() local
845 EXPECT_TRUE(IsFunctionStateful(cond_lib, *cond_func)); in TEST()
846 EXPECT_FALSE(IsFunctionStateful(cond_lib, *cond_func, true)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 430 const FunctionDef* cond_func = in IsNodeStateful() local
434 if (cond_func != nullptr) { in IsNodeStateful()
435 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *cond_func)); in IsNodeStateful()
|
| /external/llvm-project/llvm/test/Transforms/IndVarSimplify/ |
| D | eliminate-comparison.ll | 945 declare i1 @cond_func()
963 ; CHECK-NEXT: [[LOOP_COND:%.*]] = call i1 @cond_func()
989 %loop.cond = call i1 @cond_func()
1015 ; CHECK-NEXT: [[LOOP_COND:%.*]] = call i1 @cond_func()
1041 %loop.cond = call i1 @cond_func()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 661 NameAttrList cond_func; in PostprocessLiftedArgsForWhile() local
662 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "cond", &cond_func)); in PostprocessLiftedArgsForWhile()
663 const FunctionDef* cond_function_def = fld->Find(cond_func.name()); in PostprocessLiftedArgsForWhile()
667 AttrSlice(&cond_func.attr()), fld, in PostprocessLiftedArgsForWhile()
678 fld->UniqueFunctionName(absl::StrCat(cond_func.name(), "_lifted_arg_")); in PostprocessLiftedArgsForWhile()
685 &cond_func, n, fld)); in PostprocessLiftedArgsForWhile()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1776 NameAttrList cond_func; in FindArgsToLiftForWhileNode() local
1777 TF_RETURN_IF_ERROR(GetNodeAttr(while_node->def(), "cond", &cond_func)); in FindArgsToLiftForWhileNode()
1778 const FunctionDef* cond_fdef = fld->Find(cond_func.name()); in FindArgsToLiftForWhileNode()
1780 return errors::Internal("Cannot find cond function ", cond_func.name(), in FindArgsToLiftForWhileNode()
1785 *cond_fdef, AttrSlice(&cond_func.attr()), fld, &cond_fbody)); in FindArgsToLiftForWhileNode()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/ |
| D | tf_dialects.md | 73 {cond: @cond_func, true_branch: @func_foo, false_branch: @func_bar}
|