Searched refs:allocation_index (Results 1 – 9 of 9) sorted by relevance
5 // CHECK-SAME: {allocation_index = 0 : i64, offset = 128 : i64, size = 1024 : i64}9 { allocation_slice = { allocation_index = 0
292 if (info.allocation_index != -1) { in ResolveConstantGlobals()293 InsertOrDie(&globals, info.allocation_index, global); in ResolveConstantGlobals()464 &allocations_[output_info.allocation_index]; in ExecuteAsyncOnStreamImpl()467 VLOG(4) << "Looking at: allocation " << output_info.allocation_index in ExecuteAsyncOnStreamImpl()491 output_info.allocation_index); in ExecuteAsyncOnStreamImpl()526 output_info.allocation_index); in ExecuteAsyncOnStreamImpl()538 buffer_allocations.GetDeviceAddress(output_info.allocation_index); in ExecuteAsyncOnStreamImpl()614 info.allocation_index = slice.index(); in GetOutputInfo()
55 int allocation_index = -1; member60 int allocation_index; member
158 info.allocation_index = maybe_slice.ValueOrDie().index(); in EmitConstants()
1070 o.allocation_index = index; in GetMlirAllocationInfo()
794 info.allocation_index = in EmitConstant()
118 size_t allocation_index = num_lists; // heads of lists are in first num_lists nodes. in create_linked_lists() local128 pNew = &pNodes[ allocation_index++ ];// allocate a new node in create_linked_lists()
33 StructFieldAttr<"allocation_index", I64Attr>,
1185 for (int allocation_index = allocation_indices->size() - 1; in AssignSingleHloBuffer() local1186 allocation_index >= 0; allocation_index--) { in AssignSingleHloBuffer()1188 allocation_indices->at(allocation_index)); in AssignSingleHloBuffer()