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

Method memory_allocator

tensorflow/compiler/xla/service/backend.h:91–93  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

89 se::Platform* platform() const { return platform_; }
90 Compiler* compiler() const { return compiler_; }
91 se::DeviceMemoryAllocator* memory_allocator() const {
92 return memory_allocator_.get();
93 }
94 std::shared_ptr<se::DeviceMemoryAllocator> shared_memory_allocator() const {
95 return memory_allocator_;
96 }

Callers 15

AllocateMethod · 0.45
CreateAndTransferMethod · 0.45
CreateUninitializedMethod · 0.45
CreateFromBufferMethod · 0.45
SwapInMethod · 0.45
MakeTupleMethod · 0.45
RunExecutableFunction · 0.45
CompileMethod · 0.45
PopulateOutputsMethod · 0.45
RunMethod · 0.45
AllocateShapedBufferMethod · 0.45

Calls 1

getMethod · 0.45

Tested by 4

TESTFunction · 0.36
RunComputationMethod · 0.36
ExecuteAndFetchProfileFunction · 0.36
CompileAndExecuteFunction · 0.36