Searched refs:HloBufferSize (Results 1 – 2 of 2) sorted by relevance
991 << 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()
539 int64 HloBufferSize(const HloBuffer& buffer) { in HloBufferSize() function