Searched refs:is_cross_program_prefetch (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.h | 592 bool is_cross_program_prefetch = false)
598 is_cross_program_prefetch_(is_cross_program_prefetch) {}
638 bool is_cross_program_prefetch() const { in is_cross_program_prefetch() function
1272 bool is_cross_program_prefetch = false);
|
| D | hlo_instructions.cc | 210 bool is_cross_program_prefetch) in HloCopyStartInstruction() argument
212 is_cross_program_prefetch_(is_cross_program_prefetch) { in HloCopyStartInstruction()
225 if (is_cross_program_prefetch()) { in ExtraAttributesToStringImpl()
236 return is_cross_program_prefetch() == in IdenticalSlowPath()
237 casted_other.is_cross_program_prefetch(); in IdenticalSlowPath()
246 shape, new_operands[0], is_cross_program_prefetch()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.h | 599 bool is_cross_program_prefetch = false);
1900 bool is_cross_program_prefetch() const;
|
| D | memory_space_assignment_test.cc | 5147 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P() local
5149 use.instruction->is_cross_program_prefetch(); in TEST_P()
5152 is_cross_program_prefetch), in TEST_P()
5156 !use.instruction->is_cross_program_prefetch(); in TEST_P()
5207 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P() local
5209 use.instruction->is_cross_program_prefetch(); in TEST_P()
5212 is_cross_program_prefetch), in TEST_P()
5216 !use.instruction->is_cross_program_prefetch(); in TEST_P()
|
| D | hlo_instruction.cc | 175 proto.is_cross_program_prefetch()); in CreateFromProto()
1002 bool is_cross_program_prefetch) { in CreateCopyStart() argument
1004 is_cross_program_prefetch); in CreateCopyStart()
4261 bool HloInstruction::is_cross_program_prefetch() const { in is_cross_program_prefetch() function in xla::HloInstruction
4262 return Cast<HloCopyStartInstruction>(this)->is_cross_program_prefetch(); in is_cross_program_prefetch()
|
| D | hlo_instructions.h | 139 bool is_cross_program_prefetch);
141 bool is_cross_program_prefetch() const { return is_cross_program_prefetch_; } in is_cross_program_prefetch() function
|
| D | hlo.proto | 261 bool is_cross_program_prefetch = 73; field
|
| D | hlo_parser.cc | 1322 optional<bool> is_cross_program_prefetch = false; in ParseInstructionRhs() local
1324 &is_cross_program_prefetch}; in ParseInstructionRhs()
1330 shape, operands[0], *is_cross_program_prefetch)); in ParseInstructionRhs()
|
| D | memory_space_assignment.cc | 2116 AliasedOffset* aliased_offset, bool is_cross_program_prefetch) { in AddAsyncCopy() argument
2128 copy_done_schedule_before_time, is_cross_program_prefetch)); in AddAsyncCopy()
|
| D | hlo_parser_test.cc | 321 %copy-start.1 = (f32[], f32[], u32[]) copy-start(f32[] %v1), is_cross_program_prefetch=true in CreateTestCases()
|