| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_constant_sinking_test.cc | 64 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
104 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
145 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
193 auto* while_body = module->GetComputationWithName("body"); in TEST_F() local
235 auto* while_body = module->GetComputationWithName("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 | 128 auto* while_body = while_op->while_body(); in GetLoopInductionVarTupleIdx() local
250 auto* while_body = while_op->while_body(); in PatternMatchLoopTripCount() local
364 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCount() local
448 auto* while_body = while_op->while_body(); in ComputeWhileLoopTripCountUpperBound() 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_simplifier.cc | 58 HloComputation* while_body = while_op->while_body(); in TryRemoveDeadWhileParams() local
314 auto* while_body = while_op->while_body(); in TryRemoveConstantParams() local
512 auto while_body = while_op->while_body(); in TryPropagateConstant() local
646 auto* while_body = while_op->while_body(); in TryFlattenNestedTuples() local
773 auto* while_body = while_op->while_body(); in TryMergeInductionVariables() 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
221 auto* while_body = while_op->while_body(); in TEST_F() local
|
| D | while_util_test.cc | 157 HloComputation* while_body = module->GetComputationWithName("body"); in TEST() local
|
| D | while_util.cc | 260 const HloComputation& while_body) { in GetInvariantGTEsForWhileBody()
|
| D | hlo_rematerialization_test.cc | 103 HloComputation* while_cond, HloComputation* while_body, in MakeRematerializableWhileComputation()
|
| D | hlo_verifier.cc | 1355 auto* while_body = xla_while->while_body(); in HandleWhile() local
|
| D | hlo_instruction.cc | 2031 HloComputation* HloInstruction::while_body() const { in while_body() function in xla::HloInstruction
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops.py | 64 def while_body(i, *ta_list): function
|
| D | pfor.py | 886 def while_body(i, *ta_list): function
1600 def while_body(i, ta): function
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow.py | 98 def while_body(iterate_index, *state): function
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla_test.py | 146 def while_body(i): function
|