| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 75 for (const HloInstruction* instr : {while_body->parameter_instruction(0), in TryRemoveDeadWhileParams()
76 while_cond->parameter_instruction(0)}) { in TryRemoveDeadWhileParams()
100 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams()
135 operand->operand(0) != while_body->parameter_instruction(0) || in TryRemoveDeadWhileParams()
186 auto* param = comp->parameter_instruction(0); in TryRemoveDeadWhileParams()
412 while_cond->parameter_instruction(0), in TryRemoveConstantParams()
415 while_cond->parameter_instruction(0)->name()))), in TryRemoveConstantParams()
421 while_body->parameter_instruction(0), in TryRemoveConstantParams()
424 while_cond->parameter_instruction(0)->name()))), in TryRemoveConstantParams()
518 auto while_body_param = while_body->parameter_instruction(0); in TryPropagateConstant()
[all …]
|
| D | while_util.cc | 33 narrow_condition->parameter_instruction(0)->shape(); in WidenWhileCondition()
50 TupleUtil::ExtractPrefix(wide_while_cond->parameter_instruction(0), in WidenWhileCondition()
64 const Shape& narrow_shape = narrow_body->parameter_instruction(0)->shape(); in WidenWhileBody()
73 HloInstruction* wide_parameter = wide_while_body->parameter_instruction(0); in WidenWhileBody()
132 HloInstruction* while_body_param = new_while_body->parameter_instruction(0); in MakeInstructionsLiveIn()
163 HloInstruction* param = cond_computation->parameter_instruction(0); in MakeCountedLoopConditionComputation()
183 HloInstruction* param = body_computation->parameter_instruction(0); in MakeCountedLoopBodyComputation()
268 instr->operand(0) == while_body.parameter_instruction(0)) { in GetInvariantGTEsForWhileBody()
280 while_conditional.parameter_instruction(0)->users()) { in GetGTEsMapForWhileConditional()
|
| D | while_util_test.cc | 57 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
58 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
59 *param2 = (*entry_computation)->parameter_instruction(2); in GetParsedModule()
200 HloInstruction* to_make_live_in = main->parameter_instruction(1); in TEST()
|
| D | stable_sort_expander.cc | 159 replacements[comparator->parameter_instruction(i)] = in ExpandInstruction()
160 comparator->parameter_instruction(i ^ 1); in ExpandInstruction()
187 scalar_pred, comparator->parameter_instruction(2 * iota_index), in ExpandInstruction()
188 comparator->parameter_instruction(2 * iota_index + 1), in ExpandInstruction()
|
| D | dynamic_parameter_binding.cc | 119 entry->parameter_instruction(dynamic_parameter.parameter_num)->shape(), in Verify()
122 entry->parameter_instruction(dynamic_dimension.parameter_num)->shape(), in Verify()
127 entry->parameter_instruction(dynamic_dimension.parameter_num) in Verify()
|
| D | sort_simplifier.cc | 57 if (comparator->parameter_instruction(i)->user_count() > 0) { in RemoveUnusedOperandFromSort()
87 auto* old_lhs_parameter = comparator->parameter_instruction(i * 2); in RemoveUnusedOperandFromSort()
88 auto* old_rhs_parameter = comparator->parameter_instruction(i * 2 + 1); in RemoveUnusedOperandFromSort()
|
| D | while_loop_simplifier_test.cc | 563 new_while->while_body()->parameter_instruction(0)->shape(), flat_tuple)); in TEST_F()
565 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
647 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
650 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
717 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
720 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
|
| D | tuple_util_test.cc | 43 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
44 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
|
| D | bfloat16_normalization_test.cc | 215 ->parameter_instruction(0) in TEST_F()
220 ->parameter_instruction(1) in TEST_F()
329 EXPECT_EQ(sort->to_apply()->parameter_instruction(1)->shape().element_type(), in TEST_F()
333 auto users = sort->to_apply()->parameter_instruction(1)->users(); in TEST_F()
|
| D | while_loop_analysis.cc | 112 auto* while_cond_param = while_cond->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
137 auto* while_body_param = while_body->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
439 auto* while_cond_param = while_cond->parameter_instruction(0); in ComputeWhileLoopTripCountUpperBound()
|
| D | source_map_util.h | 54 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
|
| D | hlo_schedule_test.cc | 153 constant, entry->parameter_instruction(0))); in TEST_F()
264 body->set_root_instruction(body->parameter_instruction(0)); in TEST_F()
|
| D | hlo_input_output_alias_config.cc | 179 HloInstruction* param = entry->parameter_instruction(i); in Verify()
189 entry->parameter_instruction(alias.parameter_number)->shape(); in Verify()
|
| D | hlo_module_dce.cc | 49 auto* while_body_param = while_body_comp->parameter_instruction(0); in RunWhileDCE()
|
| D | defuser.cc | 57 defused_instructions[fused_computation->parameter_instruction(i)] = in Defuse()
|
| D | optimize_input_output_buffer_alias.cc | 101 return Build(module->entry_computation()->parameter_instruction(0)->shape(), in Run()
|
| D | hlo_computation_test.cc | 528 ShapeUtil::Equal(clone->parameter_instruction(0)->shape(), r0f32_)); in TEST_F()
530 ShapeUtil::Equal(clone->parameter_instruction(1)->shape(), r0f32_)); in TEST_F()
532 ShapeUtil::Equal(clone->parameter_instruction(2)->shape(), r0s32)); in TEST_F()
534 ShapeUtil::Equal(clone->parameter_instruction(3)->shape(), r0u32)); in TEST_F()
|
| D | hlo_dataflow_analysis.cc | 685 user->branch_computation(j)->parameter_instruction(0)); in Propagate()
695 called_computation->parameter_instruction(operand_number)); in Propagate()
715 callsite.instruction()->while_body()->parameter_instruction(0)); in Propagate()
717 callsite.instruction()->while_condition()->parameter_instruction( in Propagate()
|
| D | bfloat16_propagation.cc | 256 use.instruction->while_condition()->parameter_instruction( in AllUsersConsumeBF16()
262 use.instruction->while_body()->parameter_instruction( in AllUsersConsumeBF16()
414 auto parameter = computation->parameter_instruction(i); in AdjustCalledComputationParameters()
|
| D | hlo_module_dce_test.cc | 54 auto* while_body_param = while_body_comp->parameter_instruction(0); in WhileBodyHasPassThroughTupleElement()
|
| D | hlo_creation_utils_test.cc | 41 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
|
| D | while_loop_invariant_code_motion.cc | 53 HloInstruction* while_body_param = while_body->parameter_instruction(0); in CreateLoopInvariantCopy()
|
| D | dynamic_dimension_inference.cc | 537 computation->parameter_instruction(dynamic_dimension.parameter_num); in HandleParameter()
540 computation->parameter_instruction(dynamic_parameter.parameter_num); in HandleParameter()
|
| D | buffer_assignment.cc | 1548 module->entry_computation()->parameter_instruction( in BuildColocatedBufferSets()
1577 while_hlo->while_condition()->parameter_instruction(0), index, in BuildColocatedBufferSets()
1581 while_hlo->while_body()->parameter_instruction(0), index, in BuildColocatedBufferSets()
1594 const HloInstruction* call_param = callee->parameter_instruction(i); in BuildColocatedBufferSets()
1657 conditional->branch_computation(j)->parameter_instruction( in BuildColocatedBufferSets()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 73 const auto& expected_shape = computation->parameter_instruction(i)->shape(); in ExecuteOnStream()
|