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

Method on_host

tensorflow/core/kernels/cuda_solvers.h:396–396  ·  view source on GitHub ↗

Returns true if this ScratchSpace is in host memory.

Source from the content-addressed store, hash-verified

394
395 // Returns true if this ScratchSpace is in host memory.
396 bool on_host() const { return on_host_; }
397
398 protected:
399 OpKernelContext* context() const { return context_; }

Callers 15

AddRequestMethod · 0.45
RecvTensorAsyncMethod · 0.45
FuseRecvTensorAsyncMethod · 0.45
InitAllocMethod · 0.45
StartMethod · 0.45
RecvFromPeerMethod · 0.45
GrpcRecvTensorAsyncMethod · 0.45
RecvBufAsyncMethod · 0.45
RecvHandlerMethod · 0.45
AllocateTensorsAsyncMethod · 0.45
GetAllocatorLockedMethod · 0.45
MakeTensorFromProtoMethod · 0.45

Calls

no outgoing calls

Tested by 1

CheckAllocFunction · 0.36