- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreatePerDeviceValues (0.27 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TensorHandlePtr first_negative_one = CreatePerDeviceValues( context.get(), components, first_device_name, status.get()); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); components[0] = first_negative_one.get(); components[1] = negative_one_cpu.get(); TensorHandlePtr second_negative_one = CreatePerDeviceValues( context.get(), components, second_device_name, status.get());
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status); // Helper to pack `num_replicas` TFE_TensorHandles into one parallel handle. template <std::size_t num_replicas> TensorHandlePtr CreatePerDeviceValues( TFE_Context* context, const std::array<TFE_TensorHandle*, num_replicas>& components, const char* device, TF_Status* status); TensorHandlePtr Multiply(TFE_Context* context, TFE_TensorHandle* first,
C - Registered: Tue Apr 23 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_testlib.cc
std::array<TFE_TensorHandle*, 2> components{initial_value_cpu.get(), initial_value_cpu.get()}; TensorHandlePtr initial_value = CreatePerDeviceValues(context, components, device_name, status.get()); variable->Assign(context, initial_value.get(), status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
std::array<TFE_TensorHandle*, 2> in_components{value_one.get(), value_two.get()}; TensorHandlePtr combined_value = CreatePerDeviceValues( context.get(), in_components, device_name, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); // Loop to make synchronization failures more deterministic
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0)