| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_constant_sinking_test.cc | 63 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
103 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
144 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
192 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
234 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
|
| D | while_loop_expensive_invariant_code_motion_test.cc | 78 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F() local
100 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F() local
184 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F() local
234 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F() local
|
| D | while_loop_invariant_code_motion_test.cc | 66 HloComputation* while_body = [&]() { in TEST_F() local
108 HloComputation* while_body = [&]() { in TEST_F() local
180 HloComputation* while_body = [&]() { in TEST_F() local
219 HloComputation* while_body = [&]() { in TEST_F() local
258 HloComputation* while_body = [&]() { in TEST_F() local
307 HloComputation* while_body = [&]() { in TEST_F() local
361 HloComputation* while_body = [&]() { in TEST_F() local
411 HloComputation* while_body; in TEST_F() local
447 HloComputation* while_body = [&]() { in TEST_F() local
501 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F() local
[all …]
|
| D | while_loop_analysis.cc | 100 auto* while_body = while_op->while_body(); in GetAuxiliaryLoopInductionVars() local
270 auto* while_body = while_op->while_body(); in GetLoopInductionVarTupleIdx() local
392 auto* while_body = while_op->while_body(); in PatternMatchLoopTripCount() local
506 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCount() local
590 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCountUpperBound() local
|
| D | root_instruction_sinker_test.cc | 57 auto while_body = in TEST_F() local
103 auto while_body = in TEST_F() local
|
| D | while_loop_simplifier.cc | 54 HloComputation* while_body = while_op->while_body(); in RemoveDeadTupleIndices() local
226 HloComputation* while_body = while_op->while_body(); in TryRemoveDeadWhileParams() local
338 HloComputation* while_body = while_op->while_body(); in TryRemoveRepeatedWhileTupleIndicesHelper() local
400 HloComputation* while_body = while_op->while_body(); in TryRemoveRepeatedWhileTupleIndices() local
495 auto* while_body = while_op->while_body(); in TryRemoveConstantParams() local
724 auto while_body = while_op->while_body(); in TryPropagateConstant() local
858 auto* while_body = while_op->while_body(); in TryFlattenNestedTuples() local
985 auto* while_body = while_op->while_body(); in TryMergeInductionVariables() local
|
| D | while_loop_invariant_code_motion.cc | 43 HloComputation* while_body = while_instr->while_body(); in CreateLoopInvariantCopy() local
154 HloComputation* while_body = while_instr->while_body(); in TryHoistingInvariantInstructionsFromWhileBody() local
|
| D | while_loop_expensive_invariant_code_motion.cc | 57 HloComputation* while_body = while_instr->while_body(); in CreateLoopInvariantCopy() local
143 HloComputation* while_body = while_instr->while_body(); in TryHoistingInvariantInstructionsFromWhileBody() local
|
| D | while_loop_all_reduce_code_motion.cc | 67 HloComputation* while_body) { in IsAllReduceMovable()
144 HloComputation* while_body) -> BufferTupleIndex { in IsAllReduceMovable()
272 HloComputation* while_body) -> MovableAllReduceContext { in IsAllReduceMovable()
|
| D | while_loop_constant_sinking.cc | 52 HloComputation* while_body = while_instr->while_body(); in TrySinkingConstantsIntoWhileLoop() local
|
| D | while_loop_simplifier_test.cc | 187 auto* while_body = while_op->while_body(); in TEST_F() local
203 auto* while_body = while_op->while_body(); in TEST_F() local
219 auto* while_body = while_op->while_body(); in TEST_F() local
|
| D | while_util_test.cc | 162 HloComputation* while_body = module->GetComputationWithName("body"); in TEST_F() local
|
| D | while_util.cc | 275 const HloComputation& while_body) { in GetInvariantGTEsForWhileBody()
|
| D | sharding_propagation.cc | 1575 HloComputation* while_body = while_instruction->while_body(); in CheckAndUpdateDeviceAssignmentsInWhileBody() local
|
| D | memory_space_assignment.cc | 894 HloComputation* while_body = use.instruction->while_body(); in IsUseAllowedInAlternateMemory() local
1351 HloComputation* while_body = hlo_use.instruction->while_body(); in AllocateAllocationValues() local
|
| D | hlo_verifier.cc | 1817 auto* while_body = xla_while->while_body(); in HandleWhile() local
|
| D | hlo_instruction.cc | 2473 HloComputation* HloInstruction::while_body() const { in while_body() function in xla::HloInstruction
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_deprecated_py2.py | 402 def while_body(iterate_index, *loop_vars): function
466 def while_body(iterate_index, *loop_vars): function
523 def while_body(iterate, *loop_vars): function
591 def while_body(has_next, *loop_vars): function
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops.py | 70 def while_body(i, *ta_list): function
|
| D | pfor.py | 1079 def while_body(i, *ta_list): function
1999 def while_body(i, ta): function
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla_test.py | 164 def while_body(i): function
|