Searched refs:SetUpMlirAllocation (Results 1 – 3 of 3) sorted by relevance
136 static Status SetUpMlirAllocation(
886 Status GpuExecutable::SetUpMlirAllocation( in SetUpMlirAllocation() function in xla::gpu::GpuExecutable1063 TF_RETURN_IF_ERROR(SetUpMlirAllocation(func, buffer_sizes, &allocations, in LoadFromObjFile()
1605 return GpuExecutable::SetUpMlirAllocation(func, buffer_sizes, allocations, in GetMlirAllocationInfo()