| /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 | 104 auto* while_body = while_op->while_body(); in GetAuxiliaryLoopInductionVars() local
274 auto* while_body = while_op->while_body(); in GetLoopInductionVarTupleIdx() local
396 auto* while_body = while_op->while_body(); in PatternMatchLoopTripCount() local
510 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCount() local
594 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 | 69 HloComputation* while_body = while_op->while_body(); in RemoveDeadTupleIndices() local
243 HloComputation* while_body = while_op->while_body(); in TryRemoveDeadWhileParams() local
371 HloComputation* while_body = while_op->while_body(); in TryRemoveRepeatedWhileTupleIndicesHelper() local
433 HloComputation* while_body = while_op->while_body(); in TryRemoveRepeatedWhileTupleIndices() local
528 auto* while_body = while_op->while_body(); in TryRemoveConstantParams() local
762 auto while_body = while_op->while_body(); in TryPropagateConstant() local
896 auto* while_body = while_op->while_body(); in TryFlattenNestedTuples() local
1026 auto* while_body = while_op->while_body(); in TryMergeInductionVariables() local
|
| D | while_loop_invariant_code_motion.cc | 45 HloComputation* while_body = while_instr->while_body(); in CreateLoopInvariantCopy() local
158 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_expensive_invariant_code_motion.cc | 58 HloComputation* while_body = while_instr->while_body(); in CreateLoopInvariantCopy() local
144 HloComputation* while_body = while_instr->while_body(); in TryHoistingInvariantInstructionsFromWhileBody() local
|
| 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 | 303 const HloComputation& while_body) { in GetInvariantGTEsForWhileBody()
|
| D | memory_space_assignment.cc | 935 HloComputation* while_body = use.instruction->while_body(); in IsUseAllowedInAlternateMemory() local
1435 HloComputation* while_body = hlo_use.instruction->while_body(); in AllocateAllocationValues() local
3254 HloComputation* while_body = calling_instruction_->while_body(); in PostProcess() local
|
| D | sharding_propagation.cc | 1591 HloComputation* while_body = while_instruction->while_body(); in CheckAndUpdateDeviceAssignmentsInWhileBody() local
|
| D | hlo_verifier.cc | 2169 auto* while_body = xla_while->while_body(); in HandleWhile() local
|
| D | hlo_instruction.cc | 2673 HloComputation* HloInstruction::while_body() const { in while_body() function in xla::HloInstruction
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_deprecated_py2.py | 401 def while_body(iterate_index, *loop_vars): function
465 def while_body(iterate_index, *loop_vars): function
522 def while_body(iterate, *loop_vars): function
590 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 | 1090 def while_body(i, *ta_list): function
2018 def while_body(i, ta): function
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla_test.py | 164 def while_body(i): function
|