Searched refs:cond_func (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 60 AttrValue cond_func; in TEST() local
61 cond_func.mutable_func()->set_name("LessThanOrEqualToN"); in TEST()
67 .Attr("cond", cond_func) in TEST()
163 AttrValue cond_func; in TEST() local
164 cond_func.mutable_func()->set_name("XYXLessThanOrEqualToN"); in TEST()
170 .Attr("cond", cond_func) in TEST()
265 AttrValue cond_func; in TEST() local
266 cond_func.mutable_func()->set_name("LessThanOrEqualToN"); in TEST()
272 .Attr("cond", cond_func) in TEST()
|
| D | lower_if_while_test.cc | 146 AttrValue cond_func; in TEST() local
147 cond_func.mutable_func()->set_name("WhileWithIfCond"); in TEST()
153 .Attr("cond", cond_func) in TEST()
196 NameAttrList cond_func; in IfWithWhileThen() local
197 cond_func.set_name("LessThanOrEqualToN"); in IfWithWhileThen()
216 {{"cond", cond_func}, in IfWithWhileThen()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils.cc | 211 const FunctionDef* cond_func = in IsNodeStateful() local
215 if ((cond_func != nullptr && in IsNodeStateful()
216 !IsFunctionStateful(library, *cond_func, skip_assert)) && in IsNodeStateful()
|
| D | function_utils_test.cc | 831 const FunctionDef* cond_func = cond_lib.Find("__inference_test_function_19"); in TEST() local
833 EXPECT_TRUE(IsFunctionStateful(cond_lib, *cond_func)); in TEST()
834 EXPECT_FALSE(IsFunctionStateful(cond_lib, *cond_func, true)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass.cc | 187 const string cond_func = name_attr->name(); in IsCompilableWhile() local
189 call.set_op(cond_func); in IsCompilableWhile()
194 << ": can't compile loop condition: " << cond_func; in IsCompilableWhile()
|