Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc98 auto* param = comp->parameter_instruction(0); in RemoveDeadTupleIndices()
260 for (const HloInstruction* instr : {while_body->parameter_instruction(0), in TryRemoveDeadWhileParams()
261 while_cond->parameter_instruction(0)}) { in TryRemoveDeadWhileParams()
285 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams()
336 operand->operand(0) != while_body->parameter_instruction(0) || in TryRemoveDeadWhileParams()
383 comp->parameter_instruction(0)->shape().tuple_shapes(tuple_index), in TryRemoveRepeatedWhileTupleIndicesHelper()
384 comp->parameter_instruction(0), tuple_index)); in TryRemoveRepeatedWhileTupleIndicesHelper()
390 instr->operand(0) == comp->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndicesHelper()
461 pivot_body_elem->operand(0) == while_body->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndices()
480 body_elem->operand(0) == while_body->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndices()
[all …]
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.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()
138 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()
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.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()
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()
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()
Dtuple_util_test.cc52 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
53 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
Dwhile_loop_analysis.cc105 auto* while_body_param = while_body->parameter_instruction(0); in GetAuxiliaryLoopInductionVars()
258 auto* while_cond_param = while_cond->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
283 auto* while_body_param = while_body->parameter_instruction(0); in GetLoopInductionVarTupleIdx()
585 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()
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()
Doptimize_input_output_buffer_alias.cc99 return Build(module->entry_computation()->parameter_instruction(0)->shape(), in Run()
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.cc57 defused_instructions[fused_computation->parameter_instruction(i)] = in Defuse()
Dhlo_creation_utils_test.cc41 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
56 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
Dhlo_module_dce.cc52 auto* while_body_param = while_body_comp->parameter_instruction(0); in RunWhileDCE()
Dhlo_dataflow_analysis_test.cc1987 auto computation1_param = computation1->parameter_instruction(0); in TEST_P()
1988 auto computation2_param = computation2->parameter_instruction(0); in TEST_P()
1989 auto computation3_param = computation3->parameter_instruction(0); in TEST_P()
2428 auto* param = module_->entry_computation()->parameter_instruction(0); in TEST_F()
2603 HloInstruction* operand_param = computation_->parameter_instruction(0); in TEST_F()
2604 HloInstruction* indices_param = computation_->parameter_instruction(1); in TEST_F()
2605 HloInstruction* updates_param = computation_->parameter_instruction(2); in TEST_F()
2631 HloInstruction* lhs_param = computation_->parameter_instruction(0); in TEST_F()
2632 HloInstruction* rhs_param = computation_->parameter_instruction(1); in TEST_F()
2905 auto* param0 = module_->entry_computation()->parameter_instruction(0); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening.cc92 TF_RETURN_IF_ERROR(change_op_shape(condition->parameter_instruction(0))); in FlattenWhileLoop()
100 shape, condition->parameter_instruction(0), new_tuple_size - 1)); in FlattenWhileLoop()
112 TF_RETURN_IF_ERROR(change_op_shape(body->parameter_instruction(0))); in FlattenWhileLoop()
117 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()
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