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

Function TFE_EnableCollectiveOps

tensorflow/c/c_api_experimental.cc:793–804  ·  view source on GitHub ↗

Set server_def on the context, possibly updating it.

Source from the content-addressed store, hash-verified

791
792// Set server_def on the context, possibly updating it.
793TF_CAPI_EXPORT extern void TFE_EnableCollectiveOps(TFE_Context* ctx,
794 const void* proto,
795 size_t proto_len,
796 TF_Status* status) {
797 tensorflow::ServerDef server_def;
798 if (!server_def.ParseFromArray(proto, proto_len)) {
799 status->status = tensorflow::errors::InvalidArgument(
800 "Invalid tensorflow.ServerDef protocol buffer");
801 return;
802 }
803 status->status = EnableCollectiveOps(server_def, ctx);
804}
805
806std::string tensorflow::getTF_OutputDebugString(TF_Output node) {
807 return absl::Substitute("TF_Output($0, $1)", node.oper, node.index);

Callers

nothing calls this directly

Calls 2

InvalidArgumentFunction · 0.85
EnableCollectiveOpsFunction · 0.85

Tested by

no test coverage detected