- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for serial (0.16 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.cc
// TODO(allenl): We should remove "TPU" from these op names at the very least, // or consider other ways of packing/unpacking parallel tensors. if (operation_name == std::string("TPUReplicatedInput")) { // Special-cased operation for packing per-device tensors into one parallel // tensor. if (inputs.size() != parallel_device.num_underlying_devices()) { std::string message(absl::StrCat(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op.get(), "dtype", type); TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status); TFE_OpSetAttrString(op.get(), "container", "", 0); // Use the special GUID for no buffer sharing // // TODO(allenl): Should we provide a better API for this? AFAIK this is the // only reasonable way to make variables with no aliasing using the eager C // API.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0)