Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc1103 std::vector<BufferAllocation::Index>* allocation_indices, in AssignSingleHloBuffer() argument
1139 allocation_indices->push_back(allocation->index()); in AssignSingleHloBuffer()
1185 for (int allocation_index = allocation_indices->size() - 1; in AssignSingleHloBuffer()
1188 allocation_indices->at(allocation_index)); in AssignSingleHloBuffer()
1228 allocation_indices->push_back(allocation->index()); in AssignSingleHloBuffer()
1334 std::vector<BufferAllocation::Index> allocation_indices; in AssignBuffersForComputations() local
1341 &allocation_indices, assignment)); in AssignBuffersForComputations()
Dbuffer_assignment.h667 std::vector<BufferAllocation::Index>* allocation_indices,