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

Method device

tensorflow/core/framework/op_kernel.h:1234–1234  ·  view source on GitHub ↗

May be used, e.g., to get GPU handles, etc. TODO(tucker): Add example usage.

Source from the content-addressed store, hash-verified

1232 // May be used, e.g., to get GPU handles, etc.
1233 // TODO(tucker): Add example usage.
1234 DeviceBase* device() const { return params_->device; }
1235
1236 // Retrieve list of referenced tensors in out_vector. Once this is
1237 // called, it is not legal to reference any more tensors. Should

Callers 15

operator==Function · 0.45
ParamsMethod · 0.45
OpKernelContextClass · 0.45
op_kernel.ccFile · 0.45
FindKernelDefFunction · 0.45
FromProtoMethod · 0.45
MakeResourceHandleFunction · 0.45
ValidateDeviceFunction · 0.45
InstantiateNodeMethod · 0.45
PrintFunction · 0.45

Calls

no outgoing calls

Tested by 1

operator==Function · 0.36