- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for un (0.16 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.h
// parallel device creates a parallel tensor `x` with `a` on the first of // `underlying_devices` and `b` on the second. Running `a_unpacked, b_unpacked = // TPUReplicatedOutput(input=x, num_replicas=2)` un-packs the parallel tensor // into its components. // // The filled `device` struct and the allocated `device_info` struct may be // passed to TFE_RegisterCustomDevice. The `device_name` arguments must match.
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_testlib.h
// Creates a rank-one TFE_TensorHandle with value `v`. TensorHandlePtr VectorFloatTensorHandle(const std::vector<float>& v, 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,
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)