Searched refs:logical_buffer (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/ |
| H A D | hlo_proto_to_memory_visualization_utils.cc | 76 std::string GetInstructionName(const LogicalBufferProto* logical_buffer) { in GetInstructionName() argument
77 if (logical_buffer == nullptr) { in GetInstructionName()
80 if (logical_buffer->defined_at().shape_index().empty()) { in GetInstructionName()
81 return logical_buffer->defined_at().instruction_name(); in GetInstructionName()
84 logical_buffer->defined_at().instruction_name(), "{", in GetInstructionName()
85 absl::StrJoin(logical_buffer->defined_at().shape_index(), ""), "}"); in GetInstructionName()
273 for (const auto& logical_buffer : in ConvertHloProtoToPreprocessResult() local
275 VLOG(1) << "Logical buffer: " << logical_buffer.ShortDebugString(); in ConvertHloProtoToPreprocessResult()
276 id_to_logical_buffer[logical_buffer.id()] = &logical_buffer; in ConvertHloProtoToPreprocessResult()
277 all_logical_buffers.insert(&logical_buffer); in ConvertHloProtoToPreprocessResult()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_rematerialization.cc | 450 const LogicalBuffer* logical_buffer, in GetUsers() argument
458 points_to_analysis.GetBufferAliases(*logical_buffer)) { in GetUsers()
468 if (buffer_alias.instruction() != logical_buffer->instruction() && in GetUsers()
476 logical_buffer->index().size() != 1 in GetUsers()
478 : std::make_optional(logical_buffer->index().back()); in GetUsers()
665 const LogicalBuffer* logical_buffer, in CreateBufferFromLogicalBuffer() argument
668 UsesList users = GetUsers(instruction_list_, logical_buffer, in CreateBufferFromLogicalBuffer()
670 return NewBuffer(instruction_list_.GetItem(logical_buffer->instruction()), in CreateBufferFromLogicalBuffer()
671 logical_buffer->shape(), logical_buffer->index(), in CreateBufferFromLogicalBuffer()
812 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker() local
[all …]
|
| H A D | tuple_points_to_analysis.h | 221 LogicalBuffer& logical_buffer(LogicalBuffer::Id id) const { in logical_buffer() function
|
| H A D | BUILD | 1388 ":logical_buffer",
1540 ":logical_buffer",
1819 ":logical_buffer",
4013 ":logical_buffer",
4021 name = "logical_buffer",
4022 srcs = ["logical_buffer.cc"],
4023 hdrs = ["logical_buffer.h"],
4306 ":logical_buffer",
4322 ":logical_buffer",
4395 ":logical_buffer",
[all …]
|
| H A D | buffer_assignment_test.cc | 1937 for (const HloValue* logical_buffer : peak_buffers) { in TEST_F() local
1938 peak_instructions.push_back(logical_buffer->instruction()); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | BUILD | 42 "//tensorflow/compiler/xla/service:logical_buffer",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | BUILD | 475 "//tensorflow/compiler/xla/service:logical_buffer",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | BUILD | 664 "//tensorflow/compiler/xla/service:logical_buffer",
|