| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_test.cc | 3922 const auto while_loop = root->operand(0); in TEST_F() local
3974 const auto while_loop = root->operand(0); in TEST_F() local
4359 const auto while_loop = root->operand(0); in TEST_F() local
4437 const auto while_loop = root->operand(1)->operand(0); in TEST_F() local
4512 const auto while_loop = root->operand(0)->operand(0); in TEST_F() local
4603 const auto while_loop = root->operand(0); in TEST_F() local
4682 const auto while_loop = root->operand(1)->operand(0); in TEST_F() local
4759 const auto while_loop = root->operand(0)->operand(0); in TEST_F() local
4824 const auto while_loop = root->operand(0); in TEST_F() local
4963 const auto while_loop = in TEST_F() local
[all …]
|
| H A D | fft_handler.cc | 326 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in GetFinalFftUsingCollectivePermute() local
|
| H A D | spmd_partitioner.h | 578 HloInstruction* while_loop; member
|
| H A D | dot_handler.cc | 707 std::vector<ReplicaGroup> GetLoopReplicaGroups(HloInstruction* while_loop) { in GetLoopReplicaGroups()
1771 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in PartitionBaseCase() local
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | control_flow.py | 129 def while_loop( function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | control_flow_util_test.py | 77 def while_loop(x): function
|
| H A D | control_flow_ops_py_test.py | 1516 def while_loop(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | training_loop.py | 30 def while_loop(condition: Callable[..., Any], function
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | while_loop.py | 62 def while_loop(cond_fn, body_fn, carried_inputs): function
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/proto/example/ |
| D | lambda.hpp | 997 while_loop(Cond const &cond, Body const &body) in while_loop() function
1012 while_loop(Cond const &cond) in while_loop() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | while_v2.py | 71 def while_loop(cond, function
|
| H A D | control_flow_ops.py | 2528 def while_loop(cond, function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 561 while_loop = gen_xla_ops.xla_while variable
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cpu.py | 1921 def codegen_while_loop(self, while_loop): argument
|
| H A D | wrapper.py | 1981 def codegen_while_loop(self, while_loop): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | heap_simulator_test.cc | 171 HloInstruction* while_loop = in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | while_test.cc | 917 auto while_loop = [this, &body, build_condition](int count) { in XLA_TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | __init__.py | 2481 from torch._higher_order_ops import cond as cond, while_loop as while_loop unknown
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 6238 def while_loop(cond_fn, body_fn, carried_inputs, additional_inputs): function
|