Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for SetAttrType (0.1 sec)

  1. tensorflow/c/eager/gradients_internal.h

                              float value, ForwardOperation*);
    absl::Status SetAttrBool(AbstractOperation*, const char* attr_name, bool value,
                             ForwardOperation*);
    absl::Status SetAttrType(AbstractOperation*, const char* attr_name,
                             DataType value, ForwardOperation*);
    absl::Status SetAttrShape(AbstractOperation*, const char* attr_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/abstract_operation.h

      virtual absl::Status SetAttrFloat(const char* attr_name, float value) = 0;
      virtual absl::Status SetAttrBool(const char* attr_name, bool value) = 0;
      virtual absl::Status SetAttrType(const char* attr_name, DataType value) = 0;
      virtual absl::Status SetAttrShape(const char* attr_name, const int64_t* dims,
                                        const int num_dims) = 0;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        return absl::OkStatus();
      }
      absl::Status SetAttrBool(const char* attr_name, bool value) override {
        op_->node_builder.Attr(attr_name, value);
        return absl::OkStatus();
      }
      absl::Status SetAttrType(const char* const attr_name,
                               DataType value) override {
        if (!op_) {
          return absl::Status(
              absl::StatusCode::kFailedPrecondition,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental.cc

    }
    
    void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name,
                                  TF_DataType value, TF_Status* s) {
      Status status =
          unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value));
      TF_SetStatus(s, static_cast<TF_Code>(status.code()),
                   absl::StatusMessageAsCStr(status));
    }
    
    void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradients.cc

      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrBool(attr_name, value);
    }
    absl::Status SetAttrType(AbstractOperation* op_, const char* attr_name,
                             DataType value, ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrType(attr_name, value);
    }
    absl::Status SetAttrShape(AbstractOperation* op_, const char* attr_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.cc

      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name, TF_DataType value) {
      auto s = tensorflow::unwrap(op)->SetAttrType(
          attr_name, static_cast<tensorflow::DataType>(value));
      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top