Home
last modified time | relevance | path

Searched refs:parameter_instruction (Results 1 – 25 of 71) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_simplifier.cc93 auto* param = comp->parameter_instruction(0); in RemoveDeadTupleIndices()
263 for (const HloInstruction* instr : {while_body->parameter_instruction(0), in TryRemoveDeadWhileParams()
264 while_cond->parameter_instruction(0)}) { in TryRemoveDeadWhileParams()
356 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams()
367 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams()
422 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams()
436 while_body->parameter_instruction(0) && in TryRemoveDeadWhileParams()
487 comp->parameter_instruction(0)->shape().tuple_shapes(tuple_index), in TryRemoveRepeatedWhileTupleIndicesHelper()
488 comp->parameter_instruction(0), tuple_index)); in TryRemoveRepeatedWhileTupleIndicesHelper()
494 instr->operand(0) == comp->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndicesHelper()
[all …]
H A Dwhile_util.cc37 narrow_condition->parameter_instruction(0)->shape(); in WidenWhileCondition()
54 TupleUtil::ExtractPrefix(wide_while_cond->parameter_instruction(0), in WidenWhileCondition()
68 const Shape& narrow_shape = narrow_body->parameter_instruction(0)->shape(); in WidenWhileBody()
77 HloInstruction* wide_parameter = wide_while_body->parameter_instruction(0); in WidenWhileBody()
137 HloInstruction* while_body_param = new_while_body->parameter_instruction(0); in MakeInstructionsLiveIn()
170 HloInstruction* param = cond_computation->parameter_instruction(0); in MakeCountedLoopConditionComputation()
190 HloInstruction* param = body_computation->parameter_instruction(0); in MakeCountedLoopBodyComputation()
311 instr->operand(0) == while_body.parameter_instruction(0)) { in GetInvariantGTEsForWhileBody()
323 while_conditional.parameter_instruction(0)->users()) { in GetGTEsMapForWhileConditional()
H A Dhlo_replication_analysis_test.cc73 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
163 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
239 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
292 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
342 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
389 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
439 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
504 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
543 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
H A Dwhile_loop_simplifier_test.cc654 new_while->while_body()->parameter_instruction(0)->shape(), flat_tuple)); in TEST_F()
656 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
738 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
741 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
810 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
813 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
875 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
878 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
H A Dwhile_util_test.cc61 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
62 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
63 *param2 = (*entry_computation)->parameter_instruction(2); in GetParsedModule()
205 HloInstruction* to_make_live_in = main->parameter_instruction(1); in TEST_F()
H A Dstable_sort_expander.cc159 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()
H A Dhlo_replication_analysis.cc275 inst->while_condition()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
278 inst->while_condition()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
280 inst->operand(0), inst->while_body()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
283 inst->while_body()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
313 called->parameter_instruction(i)); in ComputeHloReplicationOnComputation()
323 inst->called_computations()[i]->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
402 auto param = entry->parameter_instruction(i); in ComputeHloReplication()
H A Dhlo_dataflow_analysis_test.cc1089 module_->entry_computation()->parameter_instruction(0); in TEST_P()
1169 called_computation->parameter_instruction(0); in TEST_P()
1174 called_computation->parameter_instruction(1); in TEST_P()
1915 auto computation1_param = computation1->parameter_instruction(0); in TEST_P()
1916 auto computation2_param = computation2->parameter_instruction(0); in TEST_P()
1917 auto computation3_param = computation3->parameter_instruction(0); in TEST_P()
2343 auto* param = module->entry_computation()->parameter_instruction(0); in TEST_F()
2526 HloInstruction* operand_param = computation->parameter_instruction(0); in TEST_F()
2527 HloInstruction* indices_param = computation->parameter_instruction(1); in TEST_F()
2528 HloInstruction* updates_param = computation->parameter_instruction(2); in TEST_F()
[all …]
H A Dtuple_util_test.cc49 module->entry_computation()->parameter_instruction(0); in TEST_F()
69 module->entry_computation()->parameter_instruction(0); in TEST_F()
71 module->entry_computation()->parameter_instruction(1); in TEST_F()
H A Ddynamic_parameter_binding.cc120 entry->parameter_instruction(dynamic_parameter.parameter_num)->shape(), in Verify()
123 entry->parameter_instruction(dynamic_dimension.parameter_num)->shape(), in Verify()
128 entry->parameter_instruction(dynamic_dimension.parameter_num) in Verify()
H A Dsort_simplifier.cc57 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()
H A Dhlo_creation_utils_test.cc43 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
58 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
355 MaybeMakeTuple({entry_computation->parameter_instruction(1), in TEST_F()
356 entry_computation->parameter_instruction(0)}); in TEST_F()
H A Dwhile_loop_analysis.cc112 auto* while_body_param = while_body->parameter_instruction(0); in GetAuxiliaryLoopInductionVars()
265 auto* while_cond_param = while_cond->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
290 auto* while_body_param = while_body->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
593 auto* while_cond_param = while_cond->parameter_instruction(0); in ComputeWhileLoopTripCountUpperBound()
H A Dbfloat16_normalization_test.cc217 ->parameter_instruction(0) in TEST_F()
222 ->parameter_instruction(1) in TEST_F()
389 EXPECT_EQ(sort->to_apply()->parameter_instruction(1)->shape().element_type(), in TEST_F()
393 auto users = sort->to_apply()->parameter_instruction(1)->users(); in TEST_F()
H A Dwhile_loop_concat_code_motion.cc506 element->operand(0) == body->parameter_instruction(0); in GroupHlosForConcat()
645 for (auto user : loop->while_condition()->parameter_instruction(0)->users()) { in TupleElementsUsedInCond()
719 auto param = body->parameter_instruction(0); in RewriteLoopWithConcatGroups()
720 auto cond_param = loop->while_condition()->parameter_instruction(0); in RewriteLoopWithConcatGroups()
952 auto param = body->parameter_instruction(0); in RunOnLoop()
H A Dconditional_simplifier.cc69 HloInstruction* param = computation->parameter_instruction(0); in TryRemoveUnusedConditionalOperands()
109 param = new_computation->parameter_instruction(0); in TryRemoveUnusedConditionalOperands()
147 ->parameter_instruction(0) in TryRemoveUnusedConditionalOperands()
H A Dmemory_space_assignment.cc1129 for (const HloInstruction* parameter_instruction : in FindAliases() local
1131 maybe_add_alias_with_instruction(parameter_instruction, &use); in FindAliases()
1184 while_body->parameter_instruction(0), use.operand_index); in IsUseAllowedInAlternateMemory()
1186 instruction_schedule.at(while_body->parameter_instruction(0)); in IsUseAllowedInAlternateMemory()
1239 const HloInstruction* parameter_instruction = in IsUseAllowedInAlternateMemory() local
1240 called_computation->parameter_instruction(0); in IsUseAllowedInAlternateMemory()
1243 parameter_instruction, other_use.hlo_use.operand_index); in IsUseAllowedInAlternateMemory()
1244 int64_t parameter_time = instruction_schedule.at(parameter_instruction); in IsUseAllowedInAlternateMemory()
1747 instruction_schedule.at(while_body->parameter_instruction(0)); in AllocateAllocationValues()
1755 called_computation->parameter_instruction(0))); in AllocateAllocationValues()
[all …]
H A Dbfloat16_propagation.cc281 use.instruction->while_condition()->parameter_instruction( in AllUsersConsumeBF16()
287 use.instruction->while_body()->parameter_instruction( in AllUsersConsumeBF16()
296 ->parameter_instruction(0); in AllUsersConsumeBF16()
468 auto parameter = computation->parameter_instruction(i); in AdjustCalledComputationParameters()
H A Dhlo_computation_test.cc510 ShapeUtil::Equal(clone->parameter_instruction(0)->shape(), r0f32_)); in TEST_F()
512 ShapeUtil::Equal(clone->parameter_instruction(1)->shape(), r0f32_)); in TEST_F()
514 ShapeUtil::Equal(clone->parameter_instruction(2)->shape(), r0s32)); in TEST_F()
516 ShapeUtil::Equal(clone->parameter_instruction(3)->shape(), r0u32)); in TEST_F()
H A Dhlo_input_output_alias_config.cc184 HloInstruction* param = entry->parameter_instruction(i); in Verify()
194 entry->parameter_instruction(alias.parameter_number)->shape(); in Verify()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dhlo_control_flow_flattening.cc179 TF_RETURN_IF_ERROR(change_op_shape(condition->parameter_instruction(0))); in FlattenWhileLoop()
181 replace_non_gte_users(condition->parameter_instruction(0)).status()); in FlattenWhileLoop()
197 shape, condition->parameter_instruction(0), new_tuple_size - 1)); in FlattenWhileLoop()
209 TF_RETURN_IF_ERROR(change_op_shape(body->parameter_instruction(0))); in FlattenWhileLoop()
211 replace_non_gte_users(body->parameter_instruction(0)).status()); in FlattenWhileLoop()
216 shape, body->parameter_instruction(0), new_tuple_size - 1)); in FlattenWhileLoop()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutable_base.cc85 const auto& expected_shape = computation->parameter_instruction(i)->shape(); in ExecuteAsyncOnStream()
117 const auto& expected_shape = computation->parameter_instruction(p)->shape(); in ExecuteAsyncOnStream()
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/source/fuzz/
Dtransformation_inline_function.cpp303 &parameter_index](opt::Instruction* parameter_instruction) { in AdaptInlinedInstruction() argument
306 if (*id == parameter_instruction->result_id()) { in AdaptInlinedInstruction()
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/source/fuzz/
H A Dtransformation_inline_function.cpp303 &parameter_index](opt::Instruction* parameter_instruction) { in AdaptInlinedInstruction() argument
306 if (*id == parameter_instruction->result_id()) { in AdaptInlinedInstruction()
/aosp_15_r20/external/angle/third_party/spirv-tools/src/source/fuzz/
H A Dtransformation_inline_function.cpp303 &parameter_index](opt::Instruction* parameter_instruction) { in AdaptInlinedInstruction() argument
306 if (*id == parameter_instruction->result_id()) { in AdaptInlinedInstruction()

123