Searched refs:logical_buffer (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.cc | 446 const LogicalBuffer* logical_buffer, in GetUsers() argument
454 points_to_analysis.GetBufferAliases(*logical_buffer)) { in GetUsers()
464 if (buffer_alias.instruction() != logical_buffer->instruction() && in GetUsers()
472 logical_buffer->index().size() != 1 in GetUsers()
474 : absl::make_optional(logical_buffer->index().back()); in GetUsers()
654 const LogicalBuffer* logical_buffer, in CreateBufferFromLogicalBuffer() argument
657 UsesList users = GetUsers(instruction_list_, logical_buffer, in CreateBufferFromLogicalBuffer()
659 return NewBuffer(instruction_list_.GetItem(logical_buffer->instruction()), in CreateBufferFromLogicalBuffer()
660 logical_buffer->shape(), logical_buffer->index(), in CreateBufferFromLogicalBuffer()
801 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker() local
[all …]
|
| D | tuple_points_to_analysis.h | 222 LogicalBuffer& logical_buffer(LogicalBuffer::Id id) const { in logical_buffer() function
|
| D | BUILD | 333 ":logical_buffer",
1172 ":logical_buffer",
1320 ":logical_buffer",
1599 ":logical_buffer",
3152 ":logical_buffer",
3160 name = "logical_buffer",
3161 srcs = ["logical_buffer.cc"],
3162 hdrs = ["logical_buffer.h"],
3446 ":logical_buffer",
3463 ":logical_buffer",
[all …]
|
| D | buffer_assignment_test.cc | 1901 for (const HloValue* logical_buffer : peak_buffers) { in TEST_F() local
1902 peak_instructions.push_back(logical_buffer->instruction()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | BUILD | 42 "//tensorflow/compiler/xla/service:logical_buffer",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 341 "//tensorflow/compiler/xla/service:logical_buffer",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 636 "//tensorflow/compiler/xla/service:logical_buffer",
|