Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc83 auto* param = comp->parameter_instruction(0); in RemoveDeadTupleIndices()
243 for (const HloInstruction* instr : {while_body->parameter_instruction(0), in TryRemoveDeadWhileParams()
244 while_cond->parameter_instruction(0)}) { in TryRemoveDeadWhileParams()
268 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams()
303 operand->operand(0) != while_body->parameter_instruction(0) || in TryRemoveDeadWhileParams()
350 comp->parameter_instruction(0)->shape().tuple_shapes(tuple_index), in TryRemoveRepeatedWhileTupleIndicesHelper()
351 comp->parameter_instruction(0), tuple_index)); in TryRemoveRepeatedWhileTupleIndicesHelper()
357 instr->operand(0) == comp->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndicesHelper()
428 pivot_body_elem->operand(0) == while_body->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndices()
447 body_elem->operand(0) == while_body->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndices()
[all …]
Dwhile_util.cc33 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()
133 HloInstruction* while_body_param = new_while_body->parameter_instruction(0); in MakeInstructionsLiveIn()
165 HloInstruction* param = cond_computation->parameter_instruction(0); in MakeCountedLoopConditionComputation()
185 HloInstruction* param = body_computation->parameter_instruction(0); in MakeCountedLoopBodyComputation()
283 instr->operand(0) == while_body.parameter_instruction(0)) { in GetInvariantGTEsForWhileBody()
295 while_conditional.parameter_instruction(0)->users()) { in GetGTEsMapForWhileConditional()
Dhlo_replication_analysis_test.cc74 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
215 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
268 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
318 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
365 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
415 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
480 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
519 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
549 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
576 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
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()
Dhlo_replication_analysis.cc210 inst->while_condition()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
213 inst->while_condition()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
215 inst->operand(0), inst->while_body()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
218 inst->while_body()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
247 called->parameter_instruction(i)); in ComputeHloReplicationOnComputation()
257 inst->called_computations()[i]->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
333 auto param = entry->parameter_instruction(i); in ComputeHloReplication()
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()
Dwhile_loop_simplifier_test.cc617 new_while->while_body()->parameter_instruction(0)->shape(), flat_tuple)); in TEST_F()
619 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
701 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
704 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
771 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
774 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
836 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
839 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
Ddynamic_parameter_binding.cc119 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()
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()
Dtuple_util_test.cc52 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
53 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
Dwhile_loop_analysis.cc101 auto* while_body_param = while_body->parameter_instruction(0); in GetAuxiliaryLoopInductionVars()
254 auto* while_cond_param = while_cond->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
279 auto* while_body_param = while_body->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
581 auto* while_cond_param = while_cond->parameter_instruction(0); in ComputeWhileLoopTripCountUpperBound()
Dsource_map_util.h54 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
Dhlo_schedule_test.cc152 constant, entry->parameter_instruction(0))); in TEST_F()
263 body->set_root_instruction(body->parameter_instruction(0)); in TEST_F()
Doptimize_input_output_buffer_alias.cc99 return Build(module->entry_computation()->parameter_instruction(0)->shape(), in Run()
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()
Ddefuser.cc57 defused_instructions[fused_computation->parameter_instruction(i)] = in Defuse()
Dhlo_input_output_alias_config.cc184 HloInstruction* param = entry->parameter_instruction(i); in Verify()
194 entry->parameter_instruction(alias.parameter_number)->shape(); in Verify()
Dhlo_module_dce.cc52 auto* while_body_param = while_body_comp->parameter_instruction(0); in RunWhileDCE()
Dhlo_creation_utils_test.cc41 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
56 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
Dconditional_simplifier.cc204 HloInstruction* param = computation->parameter_instruction(0); in TryRemoveUnusedConditionalOperands()
243 param = new_computation->parameter_instruction(0); in TryRemoveUnusedConditionalOperands()
281 ->parameter_instruction(0) in TryRemoveUnusedConditionalOperands()
Dbfloat16_propagation.cc285 use.instruction->while_condition()->parameter_instruction( in AllUsersConsumeBF16()
291 use.instruction->while_body()->parameter_instruction( in AllUsersConsumeBF16()
300 ->parameter_instruction(0); in AllUsersConsumeBF16()
477 auto parameter = computation->parameter_instruction(i); in AdjustCalledComputationParameters()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc85 const auto& expected_shape = computation->parameter_instruction(i)->shape(); in ExecuteAsyncOnStream()
115 const auto& expected_shape = computation->parameter_instruction(p)->shape(); in ExecuteAsyncOnStream()
/external/swiftshader/third_party/SPIRV-Tools/source/fuzz/
Dtransformation_inline_function.cpp302 &parameter_index](opt::Instruction* parameter_instruction) { in AdaptInlinedInstruction() argument
305 if (*id == parameter_instruction->result_id()) { in AdaptInlinedInstruction()
/external/deqp-deps/SPIRV-Tools/source/fuzz/
Dtransformation_inline_function.cpp302 &parameter_index](opt::Instruction* parameter_instruction) { in AdaptInlinedInstruction() argument
305 if (*id == parameter_instruction->result_id()) { in AdaptInlinedInstruction()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/fuzz/
Dtransformation_inline_function.cpp302 &parameter_index](opt::Instruction* parameter_instruction) { in AdaptInlinedInstruction() argument
305 if (*id == parameter_instruction->result_id()) { in AdaptInlinedInstruction()

123