Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dwhile_test.py46 def loop_cond(step): function
67 def loop_cond(step, rsum): function
93 def loop_cond(step, rsum): function
119 def loop_cond(step, x): function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dshape_inference_test.cc88 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
141 auto loop_cond = in TEST() local
H A Ddeadness_analysis_test.cc87 Output loop_cond; member
205 const Output& loop_cond, const Output& value) { in CreateDependentLoopInvariantValue()
222 const Output& loop_cond, int32_t value) { in CreateDependentLoopInvariantValue()
H A Dpartially_decluster_pass_test.cc529 Output loop_cond = ops::LoopCond(root.WithOpName("loop_cond"), cond); in TEST() local
H A Dresource_operation_safety_analysis_test.cc490 Output loop_cond = ops::Placeholder(root.WithOpName("init"), DT_BOOL); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_control_flow_test.cc348 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
498 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
628 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
750 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); in TEST() local
1025 auto loop_cond = in RunTest() local
H A Dfunctionalize_control_flow_util.h61 Node* loop_cond = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dforward_type_inference_test.cc119 Node* loop_cond; in TEST() local
341 Node* loop_cond; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dsaver_test.py2184 def loop_cond(it, _): function
2777 def loop_cond(it, _): function
2849 def loop_cond(it, _): function
/aosp_15_r20/external/executorch/backends/apple/mps/test/
H A Dtest_mps_models.py233 def loop_cond(accum, cnt): function
286 def loop_cond(accum, cnt): function
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dconstant_folding_test.py38 def loop_cond(idx_step, *unused_args): function
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_graph_reconstruction_test.py157 loop_cond = lambda i: math_ops.less(i, 16) function
H A Dsource_utils_test.py335 loop_cond = lambda i: math_ops.less(i, 16) function
H A Dsession_debug_testlib.py547 loop_cond = lambda i: math_ops.less(i, 16) function
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dcontrol_flow.cc37 const Node* loop_cond = nullptr; member
H A Dgraph_partition.cc520 const Edge* edge, Node* loop_cond, in AddControlLoop()
729 Node* loop_cond = cond_it->second; in AddControlFlow() local
H A Dgraph_partition_test.cc650 LoopCond loop_cond(scope, cv); in TEST() local
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py285 def loop_cond(accum, cnt): function
338 def loop_cond(accum, cnt): function
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/cli/
H A Dprofile_analyzer_cli_test.py331 loop_cond = lambda x: math_ops.less(x, 10) function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcollective_ops_test.py221 loop_cond = lambda d0, d1, i: math_ops.less(i, num_iterations) function
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dmeta_graph_test.py373 def loop_cond(it, _): function
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc282 std::vector<Node*> FindLoopExitNodes(const Node& loop_cond) { in FindLoopExitNodes()
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/
H A Dlp_bld_nir_soa.c1438 LLVMValueRef loop_cond = LLVMBuildExtractElement(gallivm->builder, cond, counter, ""); in emit_load_mem() local
1544 LLVMValueRef loop_cond = LLVMBuildExtractElement(gallivm->builder, cond, counter, ""); in emit_store_mem() local
1608 LLVMValueRef loop_cond = LLVMBuildExtractElement(gallivm->builder, cond, counter, ""); in emit_atomic_mem() local
/aosp_15_r20/art/compiler/optimizing/
H A Dload_store_elimination_test.cc1251 HInstruction* loop_cond = MakeCondition(loop, kCondLT, i_phi, n); in TEST_F() local
1296 HInstruction* loop_cond = MakeCondition(loop, kCondLT, i_phi, n); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dcontrol_flow_ops_py_test.py3058 loop_cond = lambda i, x: math_ops.less(i, 2) function
3081 loop_cond = lambda i: True function

12