Searched refs:logical_buffer (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.cc | 290 const LogicalBuffer* logical_buffer, in GetUsers() argument
298 points_to_analysis.GetBufferAliases(*logical_buffer)) { in GetUsers()
308 if (buffer_alias.instruction() != logical_buffer->instruction()) { in GetUsers()
414 const LogicalBuffer* logical_buffer, in CreateBufferFromLogicalBuffer() argument
419 ItemList users = GetUsers(instruction_list_, logical_buffer, in CreateBufferFromLogicalBuffer()
421 return NewBuffer(instruction_list_.GetItem(logical_buffer->instruction()), in CreateBufferFromLogicalBuffer()
422 size_function(logical_buffer->shape()), std::move(users), in CreateBufferFromLogicalBuffer()
523 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker() local
532 CHECK_EQ(operand_points_to.element(logical_buffer->index()).size(), 1); in MemoryUsageTracker()
534 operand_points_to.element(logical_buffer->index())[0]; in MemoryUsageTracker()
[all …]
|
| D | buffer_liveness.h | 77 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in DefaultColorer()
|
| D | BUILD | 942 ":logical_buffer",
1080 ":logical_buffer",
1126 ":logical_buffer",
1379 ":logical_buffer",
2313 ":logical_buffer",
2321 name = "logical_buffer",
2322 srcs = ["logical_buffer.cc"],
2323 hdrs = ["logical_buffer.h"],
2519 ":logical_buffer",
2535 ":logical_buffer",
[all …]
|
| D | tuple_points_to_analysis.h | 221 LogicalBuffer& logical_buffer(LogicalBuffer::Id id) const { in logical_buffer() function
|
| D | buffer_assignment_test.cc | 585 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in TEST_F()
644 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in TEST_F()
1729 for (const LogicalBuffer* logical_buffer : peak_buffers) { in TEST_F() local
1730 peak_instructions.push_back(logical_buffer->instruction()); in TEST_F()
|
| D | buffer_assignment.cc | 703 auto& buffer = points_to_analysis.logical_buffer(id); in ToProto()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | BUILD | 39 "//tensorflow/compiler/xla/service:logical_buffer",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 244 "//tensorflow/compiler/xla/service:logical_buffer",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 395 "//tensorflow/compiler/xla/service:logical_buffer",
|