Home
last modified time | relevance | path

Searched refs:is_cross_program_prefetch (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment.h632 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() function
1534 bool is_cross_program_prefetch = false);
H A Dmemory_space_assignment_test.cc6700 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P() local
6702 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() local
6781 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() local
6860 use.instruction->is_cross_program_prefetch(); in TEST_P()
[all …]
H A Dhlo_instructions.cc374 bool is_cross_program_prefetch) in HloCopyStartInstruction() argument
376 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()
H A Dhlo_instruction.h649 bool is_cross_program_prefetch = false);
2151 bool is_cross_program_prefetch() const;
H A Dhlo_instruction.cc239 proto.is_cross_program_prefetch()); in CreateFromProto()
1216 bool is_cross_program_prefetch) { in CreateCopyStart() argument
1218 is_cross_program_prefetch); in CreateCopyStart()
4887 bool HloInstruction::is_cross_program_prefetch() const { in is_cross_program_prefetch() function in xla::HloInstruction
4888 return Cast<HloCopyStartInstruction>(this)->is_cross_program_prefetch(); in is_cross_program_prefetch()
H A Dhlo_instructions.h277 bool is_cross_program_prefetch);
279 bool is_cross_program_prefetch() const { return is_cross_program_prefetch_; } in is_cross_program_prefetch() function
H A Dhlo.proto320 bool is_cross_program_prefetch = 73; field
H A Dhlo_parser.cc1700 optional<bool> is_cross_program_prefetch = false; in CreateInstruction() local
1702 &is_cross_program_prefetch}; in CreateInstruction()
1709 *shape, operands[0], *is_cross_program_prefetch)); in CreateInstruction()
H A Dmemory_space_assignment.cc2893 bool is_cross_program_prefetch) { in AddAsyncCopy() argument
2906 copy_done_schedule_before_time, is_cross_program_prefetch)); in AddAsyncCopy()
H A Dhlo_parser_test.cc356 %copy-start.1 = (f32[], f32[], u32[]) copy-start(f32[] %v1), is_cross_program_prefetch=true in CreateTestCases()