- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for instance_key (0.07 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
{{"y"}, "CollectiveReduce", {"x"}, {{"T", DT_FLOAT}, {"group_size", static_cast<int>(devices.size())}, {"group_key", 0}, {"instance_key", 0}, {"merge_op", "Add"}, {"final_op", "Id"}, {"subdiv_offsets", std::vector<int>()}}, /*dep=*/{"assert"}}, });
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TFE_OpSetAttrType(op.get(), "T", TFE_TensorHandleDataType(input)); TFE_OpSetAttrInt(op.get(), "group_size", group_size); TFE_OpSetAttrInt(op.get(), "group_key", 0); TFE_OpSetAttrInt(op.get(), "instance_key", 0); const std::string merge_op("Add"); TFE_OpSetAttrString(op.get(), "merge_op", merge_op.c_str(), merge_op.length()); const std::string final_op("Id");
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Aug 06 23:56:17 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
TF_DeleteStatus(status); TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(in)); TFE_OpSetAttrInt(op, "group_size", group_size); TFE_OpSetAttrInt(op, "group_key", 123); TFE_OpSetAttrInt(op, "instance_key", 456); TFE_OpSetAttrString(op, "merge_op", "Add", 3); TFE_OpSetAttrString(op, "final_op", "Id", 2); std::vector<int64_t> subdiv_offsets; TFE_OpSetAttrIntList(op, "subdiv_offsets", subdiv_offsets.data(),
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0)