Searched refs:is_cross_program_prefetch (Results 1 – 10 of 10) sorted by relevance
632 bool is_cross_program_prefetch = false)638 is_cross_program_prefetch_(is_cross_program_prefetch) {}684 bool is_cross_program_prefetch() const { in is_cross_program_prefetch() function1534 bool is_cross_program_prefetch = false);
6700 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P() local6702 use.instruction->is_cross_program_prefetch(); in TEST_P()6705 is_cross_program_prefetch), in TEST_P()6709 !use.instruction->is_cross_program_prefetch(); in TEST_P()6779 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P() local6781 use.instruction->is_cross_program_prefetch(); in TEST_P()6784 is_cross_program_prefetch), in TEST_P()6788 !use.instruction->is_cross_program_prefetch(); in TEST_P()6858 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P() local6860 use.instruction->is_cross_program_prefetch(); in TEST_P()[all …]
374 bool is_cross_program_prefetch) in HloCopyStartInstruction() argument376 is_cross_program_prefetch_(is_cross_program_prefetch) { in HloCopyStartInstruction()389 if (is_cross_program_prefetch()) { in ExtraAttributesToStringImpl()400 return is_cross_program_prefetch() == in IdenticalSlowPath()401 casted_other.is_cross_program_prefetch(); in IdenticalSlowPath()410 is_cross_program_prefetch()); in CloneWithNewOperandsImpl()
649 bool is_cross_program_prefetch = false);2151 bool is_cross_program_prefetch() const;
239 proto.is_cross_program_prefetch()); in CreateFromProto()1216 bool is_cross_program_prefetch) { in CreateCopyStart() argument1218 is_cross_program_prefetch); in CreateCopyStart()4887 bool HloInstruction::is_cross_program_prefetch() const { in is_cross_program_prefetch() function in xla::HloInstruction4888 return Cast<HloCopyStartInstruction>(this)->is_cross_program_prefetch(); in is_cross_program_prefetch()
277 bool is_cross_program_prefetch);279 bool is_cross_program_prefetch() const { return is_cross_program_prefetch_; } in is_cross_program_prefetch() function
320 bool is_cross_program_prefetch = 73; field
1700 optional<bool> is_cross_program_prefetch = false; in CreateInstruction() local1702 &is_cross_program_prefetch}; in CreateInstruction()1709 *shape, operands[0], *is_cross_program_prefetch)); in CreateInstruction()
2893 bool is_cross_program_prefetch) { in AddAsyncCopy() argument2906 copy_done_schedule_before_time, is_cross_program_prefetch)); in AddAsyncCopy()
356 %copy-start.1 = (f32[], f32[], u32[]) copy-start(f32[] %v1), is_cross_program_prefetch=true in CreateTestCases()