Home
last modified time | relevance | path

Searched refs:memory_space (Results 1 – 25 of 55) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_propagation.cc35 int64 memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
38 memory_space); in Run()
45 int64 memory_space = indexed_shape.shape.layout().memory_space(); in Run() local
48 instruction->fused_expression_root(), memory_space); in Run()
58 int64 memory_space) const { in Propagate()
67 if (shape->layout().memory_space() == memory_space) { in Propagate()
70 shape->mutable_layout()->set_memory_space(memory_space); in Propagate()
76 memory_space); in Propagate()
85 Propagate(position.index, parent_fusion, memory_space); in Propagate()
94 Propagate(position.index, fusion_operand, memory_space); in Propagate()
[all …]
Dmemory_space_assignment.cc69 value.shape().layout().memory_space() != in IsCrossProgramPrefetchCandidate()
709 uses() == other.uses() && memory_space() == other.memory_space() && in operator ==()
939 existing_required_assignment->memory_space == MemorySpace::kDefault) { in IsUseAllowedInAlternateMemory()
1067 if (allocation.memory_space() == MemorySpace::kAlternate) { in AppendAllocationInfoDebugString()
1448 aliased_allocation->memory_space() == MemorySpace::kAlternate) { in AllocateAllocationValues()
1502 CHECK(allocation.memory_space() == MemorySpace::kAlternate); in CreateOrAddToAliasedOffset()
1614 if (allocation->memory_space() == MemorySpace::kAlternate) { in AllocateCrossProgramPrefetchBuffer()
1678 if (aliased_allocation->memory_space() == MemorySpace::kAlternate) { in AddAliasedRequiredAssignment()
1681 AddRequiredAssignment(instruction, index, aliased_allocation->memory_space(), in AddAliasedRequiredAssignment()
1687 MemorySpaceAssignment::MemorySpace memory_space, int64 time, in AddRequiredAssignment() argument
[all …]
Dmemory_space_assignment.h47 AssignmentInformation* assignment_information_for_space(int64 memory_space) { in assignment_information_for_space() argument
49 if (space_and_info.first == memory_space) { in assignment_information_for_space()
53 assignment_info_.emplace_back(memory_space, AssignmentInformation()); in assignment_information_for_space()
524 Allocation(HloPosition defining_position, MemorySpace memory_space, in Allocation() argument
527 memory_space_(memory_space), in Allocation()
554 MemorySpace memory_space() const { return memory_space_; } in memory_space() function
589 CopyAllocation(const Allocation& prev_allocation, MemorySpace memory_space,
593 : Allocation(/*defining_position=*/{nullptr, {}}, memory_space, chunk,
1055 MemorySpaceAssignment::MemorySpace memory_space; member
1060 return memory_space == other.memory_space && offset == other.offset; in equals_ignoring_time()
[all …]
Dhlo_matchers.cc265 if (copy_done->shape().layout().memory_space() != to_space_) { in MatchAndExplain()
267 << copy_done->shape().layout().memory_space() << ", expected " in MatchAndExplain()
280 if (copy_start_operand->shape().layout().memory_space() != from_space_) { in MatchAndExplain()
282 << copy_start_operand->shape().layout().memory_space() in MatchAndExplain()
Dmemory_space_propagation.h38 int64 memory_space) const;
Dhlo_cost_analysis.cc1101 absl::optional<int64> memory_space) const { in GetBytesRead()
1109 index_memory_space = indexed_shape.shape.layout().memory_space(); in GetBytesRead()
1111 if (!memory_space || memory_space == index_memory_space) { in GetBytesRead()
1121 const HloInstruction& hlo, absl::optional<int64> memory_space) const { in GetBytesWritten()
1127 index_memory_space = indexed_shape.shape.layout().memory_space(); in GetBytesWritten()
1129 if (!memory_space || memory_space == index_memory_space) { in GetBytesWritten()
Dmemory_space_assignment_test.cc116 subshape.layout().memory_space() == kAlternateMemorySpace) { in AssignMemorySpace()
170 EXPECT_EQ(subshape.layout().memory_space(), kAlternateMemorySpace) in CheckPresetAssignments()
186 EXPECT_NE(subshape.layout().memory_space(), kAlternateMemorySpace) in CheckParametersInDefaultMemory()
213 .memory_space() == kAlternateMemorySpace) { in CountMaximumOutstandingAsyncCopies()
220 if (instruction->shape().layout().memory_space() == in CountMaximumOutstandingAsyncCopies()
917 EXPECT_EQ(bitcast->shape().layout().memory_space(), kAlternateMemorySpace); in TEST_P()
953 EXPECT_EQ(add->operand(0)->shape().layout().memory_space(), in TEST_P()
1012 EXPECT_EQ(add->operand(0)->shape().layout().memory_space(), in TEST_P()
1014 EXPECT_EQ(add->shape().layout().memory_space(), kAlternateMemorySpace); in TEST_P()
1017 EXPECT_EQ(mul->operand(0)->shape().layout().memory_space(), in TEST_P()
[all …]
Dhlo_cost_analysis.h171 absl::optional<int64> memory_space = absl::nullopt) const;
174 absl::optional<int64> memory_space = absl::nullopt) const;
/external/tensorflow/tensorflow/compiler/xla/
Dlayout.cc59 layout.set_memory_space(proto.memory_space()); in CreateFromProto()
87 if (memory_space() != 0) { in ToString()
88 absl::StrAppend(&colon_string, "S(", memory_space(), ")"); in ToString()
112 if (!ignore_memory_space_ && lhs.memory_space() != rhs.memory_space()) { in operator ()()
Dlayout.h94 int64 element_size_in_bits = 0, int64 memory_space = 0)
99 memory_space_(memory_space) {} in format_()
213 int64 memory_space() const { return memory_space_; } in memory_space() function
Dlayout_util.cc60 int64 element_size_in_bits, int64 memory_space) { in MakeLayout() argument
77 layout.set_memory_space(memory_space); in MakeLayout()
457 hash_value = Hash64Combine(hash_value, layout.memory_space()); in Hash()
Dlayout_util.h42 int64 memory_space = 0);
/external/tensorflow/tensorflow/core/profiler/convert/
Dop_metrics_db_combiner.cc85 dst_memory_accessed_map[{dst_memory_accessed.memory_space(), in CombineMemoryAccessedBreakdown()
90 uint64 memory_space = src_memory_accessed.memory_space(); in CombineMemoryAccessedBreakdown() local
93 dst_memory_accessed_map[{memory_space, operation_type}]; in CombineMemoryAccessedBreakdown()
96 dst_memory_accessed->set_memory_space(memory_space); in CombineMemoryAccessedBreakdown()
/external/tensorflow/tensorflow/stream_executor/
Dtf_allocator_adapter.h44 int64 memory_space) override;
83 int64 memory_space) override { in Allocate() argument
86 device_ordinal, size, retry_on_failure, memory_space); in Allocate()
Ddevice_memory_allocator.h179 int64 memory_space) = 0;
203 int64 memory_space = 0) {
204 return Allocate(device_ordinal, size, retry_on_failure, memory_space);
246 int64 memory_space) override;
Dtf_allocator_adapter.cc40 int64 memory_space) { in Allocate() argument
41 CHECK_EQ(memory_space, 0); in Allocate()
Dstream_executor_pimpl.h124 DeviceMemory<T> AllocateArray(uint64 element_count, int64 memory_space = 0);
546 DeviceMemoryBase Allocate(uint64 size, int64 memory_space);
811 int64 memory_space) { in AllocateArray() argument
813 return DeviceMemory<T>(Allocate(bytes, memory_space)); in AllocateArray()
Dstream_executor_pimpl.cc487 DeviceMemoryBase StreamExecutor::Allocate(uint64 size, int64 memory_space) { in Allocate() argument
496 DeviceMemoryBase buf = implementation_->Allocate(size, memory_space); in Allocate()
498 << ", memory_space=" << memory_space << ") returns " << buf.opaque() in Allocate()
899 int64 memory_space) { in Allocate() argument
902 DeviceMemoryBase result = executor->AllocateArray<uint8>(size, memory_space); in Allocate()
/external/llvm-project/polly/lib/External/ppcg/
Dgpu_print.c83 const char *memory_space) in gpu_array_info_print_declaration_argument() argument
92 if (memory_space) { in gpu_array_info_print_declaration_argument()
93 p = isl_printer_print_str(p, memory_space); in gpu_array_info_print_declaration_argument()
Dgpu_print.h19 const char *memory_space);
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h104 bool retry_on_failure, int64_t memory_space,
215 int64_t memory_space; member
Dc_api_conversions.cc91 bool retry_on_failure, int64_t memory_space, in ToC()
96 ->Allocate(device_ordinal, size, retry_on_failure, memory_space); in ToC()
274 c_layout->memory_space = layout.memory_space(); in ToC()
289 c_layout->memory_space); in FromC()
Dtpu_executor_interface.h47 CreateTemporaryDeviceMemory(int64 memory_space, int64 byte_offset, in CreateTemporaryDeviceMemory() argument
Dtpu_executor.h69 DeviceMemoryBase Allocate(uint64 size, int64 memory_space) override;
170 CreateTemporaryDeviceMemory(int64 memory_space, int64 byte_offset, in CreateTemporaryDeviceMemory() argument
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc47 int64 memory_space) { in Allocate() argument
55 device_ordinal, size, retry_on_failure, memory_space); in Allocate()

123