MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetAllocator

Method GetAllocator

tensorflow/compiler/jit/xla_device.cc:287–290  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

285}
286
287Allocator* XlaDevice::GetAllocator(AllocatorAttributes attr) {
288 mutex_lock lock(mu_);
289 return GetAllocatorLocked(attr);
290}
291
292Allocator* XlaDevice::GetAllocatorLocked(AllocatorAttributes attr) {
293 if (attr.on_host()) {

Callers 15

InitAllocMethod · 0.45
ComputeMethod · 0.45
GrpcRecvTensorAsyncMethod · 0.45
TransportOptionsFromTensorMethod · 0.45
TensorFromTransportOptionsMethod · 0.45
CloneMethod · 0.45
AllocateTensorsMethod · 0.45
ScopedDeviceMemoryMethod · 0.45
PopulateOutputsMethod · 0.45
CreateKernelMethod · 0.45
CompileMethod · 0.45
ComputeMethod · 0.45

Calls

no outgoing calls

Tested by 3

CheckNumberOfAllocationsMethod · 0.36
TEST_FFunction · 0.36
GetOptimizedModuleMethod · 0.36