- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TPUReplicatedInput (0.36 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.h
// the documentation for TFE_RegisterCustomDevice for more information). // // Tensors may be copied on or off the device explicitly using // TPUReplicatedInput and TPUReplicatedOutput respectively. For example, with // two component devices, running `x = TPUReplicatedInput(inputs=[a, b])` on the // parallel device creates a parallel tensor `x` with `a` on the first of
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jun 04 21:49:16 GMT 2020 - 2.9K bytes - Viewed (0) -
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.h
TFE_Context* context, const std::array<TFE_TensorHandle*, num_replicas>& components, const char* device, TF_Status* status) { std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context, "TPUReplicatedInput", status), TFE_DeleteOp); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrInt(op.get(), "N", num_replicas); for (int i = 0; i < num_replicas; ++i) {
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
ASSERT_TRUE(TF_GetCode(status.get()) == TF_INVALID_ARGUMENT) << TF_Message(status.get()); } { // Try to pass a ParallelTensor to TPUReplicatedInput std::array<TFE_TensorHandle*, 1> correct_components{value_one.get()}; TensorHandlePtr combined_value = CreatePerDeviceValues( context.get(), correct_components, device_name, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1)