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

Method set_on_host

tensorflow/core/framework/allocator.h:305–305  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

303// Allocator* a = allocator(attr);
304struct AllocatorAttributes {
305 void set_on_host(bool v) { value |= (static_cast<int>(v)); }
306 bool on_host() const { return value & 0x1; }
307 void set_nic_compatible(bool v) { value |= (static_cast<int>(v) << 1); }
308 bool nic_compatible() const { return value & (0x1 << 1); }

Callers 15

GrpcRecvTensorAsyncMethod · 0.80
CloneMethod · 0.80
RunComposedOpMethod · 0.80
CompileMethod · 0.80
ComputeMethod · 0.80
ComputeMethod · 0.80
ComputeMethod · 0.80
TESTFunction · 0.80
ComputeMethod · 0.80
ComputeAsyncMethod · 0.80

Calls

no outgoing calls

Tested by 5

RunComposedOpMethod · 0.64
TESTFunction · 0.64
SetOutputAttrsMethod · 0.64
SetOutputAttrsFunction · 0.64
CreateOpKernelContextMethod · 0.64