Home
last modified time | relevance | path

Searched defs:loop_cond (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dwhile_test.py50 def loop_cond(step): function
71 def loop_cond(step, rsum): function
97 def loop_cond(step, rsum): function
123 def loop_cond(step, x): function
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc88 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
141 auto loop_cond = in TEST() local
Ddeadness_analysis_test.cc87 Output loop_cond; member
204 const Output& loop_cond, const Output& value) { in CreateDependentLoopInvariantValue()
221 const Output& loop_cond, int32 value) { in CreateDependentLoopInvariantValue()
Dresource_operation_safety_analysis_test.cc490 Output loop_cond = ops::Placeholder(root.WithOpName("init"), DT_BOOL); in TEST() local
Dmark_for_compilation_pass_test.cc1681 Output loop_cond = ops::LoopCond(root.WithOpName("loop_cond"), cond); in TEST() local
Dextract_outside_compilation_pass.cc1504 Node* loop_cond; in AddSendLoopPredToLoopCond() local
/external/tensorflow/tensorflow/python/grappler/
Dconstant_folding_test.py42 def loop_cond(idx_step, *unused_args): function
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_graph_reconstruction_test.py161 loop_cond = lambda i: math_ops.less(i, 16) function
Dsource_utils_test.py339 loop_cond = lambda i: math_ops.less(i, 16) function
Dsession_debug_testlib.py552 loop_cond = lambda i: math_ops.less(i, 16) function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.h61 Node* loop_cond = nullptr; member
Dfunctionalize_control_flow_test.cc329 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
479 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
609 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
731 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
1006 auto loop_cond = in RunTest() local
/external/tensorflow/tensorflow/core/graph/
Dcontrol_flow.cc37 const Node* loop_cond = nullptr; member
Dgraph_partition.cc521 const Edge* edge, Node* loop_cond, in AddControlLoop()
730 Node* loop_cond = cond_it->second; in AddControlFlow() local
Dgraph_partition_test.cc649 LoopCond loop_cond(scope, cv); in TEST() local
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py335 loop_cond = lambda x: math_ops.less(x, 10) function
/external/tensorflow/tensorflow/python/training/
Dsaver_test.py2118 def loop_cond(it, _): function
2711 def loop_cond(it, _): function
2783 def loop_cond(it, _): function
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_test.py225 loop_cond = lambda d0, d1, i: math_ops.less(i, num_iterations) function
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py377 def loop_cond(it, _): function
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc281 std::vector<Node*> FindLoopExitNodes(const Node& loop_cond) { in FindLoopExitNodes()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc878 void Print(LoopCondOp loop_cond, OpAsmPrinter &p) { in Print()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py3043 loop_cond = lambda i, x: math_ops.less(i, 2) function
3066 loop_cond = lambda i: True function