Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for getType (0.17 sec)

  1. tensorflow/c/eager/abstract_op_attrs.h

      virtual bool GetFloat(absl::string_view attr_name, float* result) const = 0;
      virtual bool GetBool(absl::string_view attr_name, bool* result) const = 0;
      virtual bool GetType(absl::string_view attr_name, DataType* result) const = 0;
      virtual Status GetTypeList(
          absl::string_view attr_name,
          absl::InlinedVector<DataType, 4>* type_list) const = 0;
    
     private:
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed May 26 22:20:27 GMT 2021
    - 2K bytes
    - Viewed (0)
  2. 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)
  3. 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)
Back to top