| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_constant_sinking_test.cc | 63 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
110 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
150 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
191 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
239 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
281 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 | 111 auto* while_body = while_op->while_body(); in GetAuxiliaryLoopInductionVars() local
281 auto* while_body = while_op->while_body(); in GetLoopInductionVarTupleIdx() local
403 auto* while_body = while_op->while_body(); in PatternMatchLoopTripCount() local
517 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCount() local
602 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 | 64 HloComputation* while_body = while_op->while_body(); in RemoveDeadTupleIndices() local
239 HloComputation* while_body = while_op->while_body(); in TryRemoveDeadWhileParams() local
475 HloComputation* while_body = while_op->while_body(); in TryRemoveRepeatedWhileTupleIndicesHelper() local
537 HloComputation* while_body = while_op->while_body(); in TryRemoveRepeatedWhileTupleIndices() local
632 auto* while_body = while_op->while_body(); in TryRemoveConstantParams() local
867 auto while_body = while_op->while_body(); in TryPropagateConstant() local
1001 auto* while_body = while_op->while_body(); in TryFlattenNestedTuples() local
1132 auto* while_body = while_op->while_body(); in TryMergeInductionVariables() local
|
| D | while_loop_invariant_code_motion.cc | 46 HloComputation* while_body = while_instr->while_body(); in CreateLoopInvariantCopy() local
159 HloComputation* while_body = while_instr->while_body(); in TryHoistingInvariantInstructionsFromWhileBody() local
|
| 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_all_reduce_code_motion.cc | 159 HloInstruction* all_reduce, HloComputation* while_body, in IsAllReduceMovable()
278 HloComputation* while_body) -> BufferTupleIndex { in IsAllReduceMovable()
464 HloComputation* while_body) -> MovableAllReduceContext { in IsAllReduceMovable()
|
| D | while_loop_constant_sinking.cc | 70 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 | 1177 HloComputation* while_body = use.instruction->while_body(); in IsUseAllowedInAlternateMemory() local
1737 HloComputation* while_body = hlo_use.instruction->while_body(); in AllocateAllocationValues() local
3760 HloComputation* while_body = calling_instruction_->while_body(); in PostProcess() local
|
| D | sharding_propagation.cc | 865 HloComputation* while_body = while_instruction->while_body(); in CheckAndUpdateDeviceAssignmentsInWhileBody() local
|
| D | hlo_verifier.cc | 2398 auto* while_body = xla_while->while_body(); in HandleWhile() local
|
| D | hlo_evaluator.cc | 467 HloComputation* while_body = while_op->while_body(); in PatternMatchParseWhileLoop() local
|
| D | hlo_instruction.cc | 2834 HloComputation* HloInstruction::while_body() const { in while_body() function in xla::HloInstruction
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | convert_control_to_data_outputs.cc | 376 func::FuncOp while_body, SmallVectorImpl<TF::WhileOp>& while_callers, in ConvertControlToDataOutputs()
476 func::FuncOp while_body = entry.getFirst(); in runOnOperation() local
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops.py | 69 def while_body(i, *ta_list): function
|
| D | pfor.py | 1087 def while_body(i, *ta_list): function
2017 def while_body(i, ta): function
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla_test.py | 160 def while_body(i): function
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening_test.cc | 788 auto* while_body = outer_while->while_body(); in TEST_F() local
|