- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for set_dtype (0.15 sec)
-
tensorflow/c/eager/tape.h
sources.reserve(output_tensors.size()); for (const TapeTensor& output_tensor : output_tensors) { // 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 {
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) -
tensorflow/c/eager/gradients.h
class TapeTensor { public: explicit TapeTensor(AbstractTensorHandle* handle); TapeTensor(const TapeTensor& other); ~TapeTensor(); int64_t GetID() const; tensorflow::DataType GetDType() const; AbstractTensorHandle* ZerosLike() const; AbstractTensorHandle* GetHandle() const; private: AbstractTensorHandle* handle_; }; // A tracing/immediate-execution agnostic tape.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0)