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

Function TF_SetDevice

tensorflow/c/c_api.cc:787–789  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

785}
786
787void TF_SetDevice(TF_OperationDescription* desc, const char* device) {
788 desc->node_builder.Device(device);
789}
790
791void TF_AddInput(TF_OperationDescription* desc, TF_Output input) {
792 desc->node_builder.Input(&input.oper->node, input.index);

Callers 5

UseHelperMethod · 0.85
TEST_FFunction · 0.85
BinaryOpHelperFunction · 0.85
Split3HelperFunction · 0.85

Calls 1

DeviceMethod · 0.45

Tested by 4

UseHelperMethod · 0.68
TEST_FFunction · 0.68
BinaryOpHelperFunction · 0.68
Split3HelperFunction · 0.68