Home
last modified time | relevance | path

Searched refs:logical_buffer (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc446 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 …]
Dtuple_points_to_analysis.h222 LogicalBuffer& logical_buffer(LogicalBuffer::Id id) const { in logical_buffer() function
DBUILD333 ":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 …]
Dbuffer_assignment_test.cc1901 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/
DBUILD42 "//tensorflow/compiler/xla/service:logical_buffer",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD341 "//tensorflow/compiler/xla/service:logical_buffer",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD636 "//tensorflow/compiler/xla/service:logical_buffer",