- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for variant (0.27 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.h
#include <memory> #include <optional> #include <string> #include <utility> #include <vector> #include "absl/types/optional.h" #include "absl/types/span.h" #include "absl/types/variant.h" #include "tensorflow/c/c_api.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/c/eager/tfe_op_internal.h" #include "tensorflow/c/safe_ptr.h"
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
// Ownership of `aid` transferred to CallBackwardFunction below. Gradient* aid; if (output_tensor.GetDType() == tensorflow::DT_VARIANT) { // Note: Needs to be zeros rather than ones since there's currently no // ones_like for variants. aid = output_tensor.ZerosLike(); } else { // TODO(allenl): Figure out why using zeros_like everywhere causes issues
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1)