Home
last modified time | relevance | path

Searched refs:allocation_index (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc306 if (info.allocation_index != -1) { in ResolveConstantGlobals()
307 InsertOrDie(&globals, info.allocation_index, global); in ResolveConstantGlobals()
649 &allocations_[output_info.allocation_index]; in ExecuteAsyncOnStreamImpl()
652 VLOG(4) << "Looking at: allocation " << output_info.allocation_index in ExecuteAsyncOnStreamImpl()
676 output_info.allocation_index); in ExecuteAsyncOnStreamImpl()
712 output_info.allocation_index); in ExecuteAsyncOnStreamImpl()
724 buffer_allocations.GetDeviceAddress(output_info.allocation_index); in ExecuteAsyncOnStreamImpl()
829 info.allocation_index = slice.index(); in GetOutputInfo()
Dgpu_executable.h59 int allocation_index = -1; member
64 int allocation_index; member
Dgpu_compiler.cc1249 o.allocation_index = i; in GetMlirAllocationInfo()
Dir_emitter_unnested.cc625 info.allocation_index = in EmitConstant()
/external/OpenCL-CTS/test_conformance/SVM/
Dmain.cpp118 size_t allocation_index = num_lists; // heads of lists are in first num_lists nodes. in create_linked_lists() local
128 pNew = &pNodes[ allocation_index++ ];// allocate a new node in create_linked_lists()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc1198 for (int allocation_index = allocation_indices->size() - 1; in AssignSingleHloBuffer() local
1199 allocation_index >= 0; allocation_index--) { in AssignSingleHloBuffer()
1201 allocation_indices->at(allocation_index)); in AssignSingleHloBuffer()