Home
last modified time | relevance | path

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

/external/mesa3d/src/intel/compiler/elk/
Delk_predicated_break.cpp222 elk_backend_instruction *while_inst = while_block->start(); in elk_opt_predicated_break() local
225 while_inst->opcode == ELK_OPCODE_WHILE && in elk_opt_predicated_break()
226 while_inst->predicate == ELK_PREDICATE_NONE && in elk_opt_predicated_break()
229 while_inst->predicate = jump_inst->predicate; in elk_opt_predicated_break()
230 while_inst->predicate_inverse = !jump_inst->predicate_inverse; in elk_opt_predicated_break()
Delk_eu_emit.c1650 elk_patch_break_cont(struct elk_codegen *p, elk_inst *while_inst) in elk_patch_break_cont() argument
1659 for (inst = while_inst - 1; inst != do_inst; inst--) { in elk_patch_break_cont()
1666 elk_inst_set_gfx4_jump_count(devinfo, inst, br*((while_inst - inst) + 1)); in elk_patch_break_cont()
1669 elk_inst_set_gfx4_jump_count(devinfo, inst, br * (while_inst - inst)); in elk_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_utils.h115 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
118 ShapeUtil::MakeShape(xla::F32, {1025}), {bcast, while_inst},
Dlayout_assignment.cc863 Status CheckWhileLayout(HloInstruction* while_inst, in CheckWhileLayout() argument
866 auto init_shape = while_inst->operand(0)->shape(); in CheckWhileLayout()
874 TF_RET_CHECK(LayoutsInShapesEqual(init_shape, while_inst->shape())); in CheckWhileLayout()
/external/igt-gpu-tools/assembler/
Dbrw_eu_emit.c1611 brw_patch_break_cont(struct brw_compile *p, struct brw_instruction *while_inst) in brw_patch_break_cont() argument
1618 for (inst = while_inst - 1; inst != do_inst; inst--) { in brw_patch_break_cont()
1625 inst->bits3.if_else.jump_count = br * ((while_inst - inst) + 1); in brw_patch_break_cont()
1628 inst->bits3.if_else.jump_count = br * (while_inst - inst); in brw_patch_break_cont()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc9824 const HloInstruction* while_inst = FindInstruction(module.get(), "while"); in TEST_F() local
9825 EXPECT_NE(while_inst, nullptr); in TEST_F()
9826 const HloComputation* cond_comp = while_inst->while_condition(); in TEST_F()
9859 const HloInstruction* while_inst = FindInstruction(module.get(), "while"); in TEST_F() local
9860 EXPECT_NE(while_inst, nullptr); in TEST_F()
9861 const HloComputation* cond_comp = while_inst->while_condition(); in TEST_F()
9906 const HloInstruction* while_inst = FindInstruction(module.get(), "while"); in TEST_F() local
9907 EXPECT_NE(while_inst, nullptr); in TEST_F()
9908 const HloComputation* cond_comp = while_inst->while_condition(); in TEST_F()
9953 const HloInstruction* while_inst = FindInstruction(module.get(), "while"); in TEST_F() local
[all …]