Home
last modified time | relevance | path

Searched refs:while_inst (Results 1 – 5 of 5) sorted by relevance

/external/mesa3d/src/intel/compiler/
Dbrw_predicated_break.cpp122 backend_instruction *while_inst = while_block->start(); in opt_predicated_break() local
125 while_inst->opcode == BRW_OPCODE_WHILE && in opt_predicated_break()
126 while_inst->predicate == BRW_PREDICATE_NONE) { in opt_predicated_break()
128 while_inst->predicate = jump_inst->predicate; in opt_predicated_break()
129 while_inst->predicate_inverse = !jump_inst->predicate_inverse; in opt_predicated_break()
Dbrw_eu_emit.c1720 brw_patch_break_cont(struct brw_codegen *p, brw_inst *while_inst) in brw_patch_break_cont() argument
1729 for (inst = while_inst - 1; inst != do_inst; inst--) { in brw_patch_break_cont()
1736 brw_inst_set_gen4_jump_count(devinfo, inst, br*((while_inst - inst) + 1)); in brw_patch_break_cont()
1739 brw_inst_set_gen4_jump_count(devinfo, inst, br * (while_inst - inst)); in brw_patch_break_cont()
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc199 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
209 EXPECT_THAT(while_inst->while_body()->instructions(), Contains(op::Add())); in TEST_F()
239 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
248 EXPECT_THAT(while_inst->while_body()->instructions(), Contains(op::Add())); in TEST_F()
282 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
286 HloInstruction::CreateGetTupleElement(scalar_s32, while_inst, 0)); in TEST_F()
293 EXPECT_THAT(while_inst->while_body()->instructions(), in TEST_F()
335 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
339 HloInstruction::CreateGetTupleElement(scalar_s32, while_inst, 0)); in TEST_F()
347 EXPECT_THAT(while_inst->while_body()->instructions(), in TEST_F()
[all …]
Dhlo_rematerialization_test.cc116 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in MakeRematerializableWhileComputation() local
119 ShapeUtil::MakeShape(xla::F32, {1025}), {bcast, while_inst}, in MakeRematerializableWhileComputation()
Dlayout_assignment.cc691 Status CheckWhileLayout(HloInstruction* while_inst, in CheckWhileLayout() argument
694 auto init_shape = while_inst->operand(0)->shape(); in CheckWhileLayout()
703 LayoutUtil::LayoutsInShapesEqual(init_shape, while_inst->shape())); in CheckWhileLayout()