- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ExtractPerDeviceValues (0.22 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status)); } // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle. template <std::size_t num_replicas> void ExtractPerDeviceValues( TFE_Context* context, TFE_TensorHandle* input, std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status) { std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op(
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) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
// successful. The resulting structure should be: // second_device{first_device{1. * 3., 2. * 3.}, 3. * 3.}. std::array<TensorHandlePtr, 2> second_components; ExtractPerDeviceValues(context.get(), multiply_result.get(), &second_components, status.get()); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(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) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
status.get())); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); std::array<TensorHandlePtr, 2> out_components; ExtractPerDeviceValues(context.get(), multiply_result.get(), &out_components, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
TF_Status* status); // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle. template <std::size_t num_replicas> void ExtractPerDeviceValues( TFE_Context* context, TFE_TensorHandle* input, std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status); // Helper to pack `num_replicas` TFE_TensorHandles into one parallel handle.
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)