Home
last modified time | relevance | path

Searched refs:HloBufferSize (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc991 << assignment->HloBufferSize(hlo_buffer) in MaybeAssignBuffer()
1000 if (assignment->HloBufferSize(hlo_buffer) > allocation->size()) { in MaybeAssignBuffer()
1002 << assignment->HloBufferSize(hlo_buffer) << " > " in MaybeAssignBuffer()
1087 allocation->size() != assignment->HloBufferSize(hlo_buffer)) { in MaybeAssignBuffer()
1094 assignment->HloBufferSize(hlo_buffer)); in MaybeAssignBuffer()
1105 const int64 buffer_size = assignment->HloBufferSize(*hlo_buffer); in AssignSingleHloBuffer()
1313 const int64 a_size = assignment->HloBufferSize(*a); in AssignBuffersForComputations()
1314 const int64 b_size = assignment->HloBufferSize(*b); in AssignBuffersForComputations()
Dbuffer_assignment.h539 int64 HloBufferSize(const HloBuffer& buffer) { in HloBufferSize() function