Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
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 …]
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()
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()
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()
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.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()
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()
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()
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()
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()
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()
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 …]
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()
Dsource_map_util.h53 HloInstruction* param = computation.parameter_instruction(parameter_number); in InvalidParameterArgument()
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()
Dhlo_input_output_alias_config.cc184 HloInstruction* param = entry->parameter_instruction(i); in Verify()
194 entry->parameter_instruction(alias.parameter_number)->shape(); in Verify()
Ddefuser.cc56 defused_instructions[fused_computation->parameter_instruction(i)] = in Defuse()
Doptimize_input_output_buffer_alias.cc101 return Build(module->entry_computation()->parameter_instruction(0)->shape(), in Run()
Dreal_imag_expander_test.cc117 auto param = module->entry_computation()->parameter_instruction(0); in TEST_F()
Dhlo_schedule_test.cc153 constant, entry->parameter_instruction(0))); in TEST_F()
264 body->set_root_instruction(body->parameter_instruction(0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
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()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
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()
/external/swiftshader/third_party/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()
/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.cpp303 &parameter_index](opt::Instruction* parameter_instruction) { in AdaptInlinedInstruction() argument
306 if (*id == parameter_instruction->result_id()) { in AdaptInlinedInstruction()

123