- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for CollectiveReduce (0.55 sec)
-
tensorflow/c/eager/c_api_test_util.cc
return op; } TFE_Op* AllReduceOp(TFE_Context* ctx, TFE_TensorHandle* in, int group_size) { TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "CollectiveReduce", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(op, in, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TF_DeleteStatus(status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0)