Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 5 of 5 for SetAttrShape (0.21 seconds)

  1. tensorflow/c/eager/abstract_operation.h

      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;
      virtual absl::Status SetAttrShape(const char* attr_name,
                                        const PartialTensorShape shape);
      virtual absl::Status SetAttrFunction(const char* attr_name,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 7.3K bytes
    - Click Count (0)
  2. tensorflow/c/eager/gradients_internal.h

                             ForwardOperation*);
    absl::Status SetAttrType(AbstractOperation*, const char* attr_name,
                             DataType value, ForwardOperation*);
    absl::Status SetAttrShape(AbstractOperation*, const char* attr_name,
                              const int64_t* dims, const int num_dims,
                              ForwardOperation*);
    absl::Status SetAttrFunction(AbstractOperation*, const char* attr_name,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 4.7K bytes
    - Click Count (0)
  3. tensorflow/c/eager/c_api_unified_experimental_graph.cc

              "op_type and op_name must be specified before specifying attrs.");
        }
        op_->node_builder.Attr(attr_name, value);
        return absl::OkStatus();
      }
      absl::Status SetAttrShape(const char* attr_name, const int64_t* dims,
                                const int num_dims) override {
        PartialTensorShape shape;
        if (num_dims >= 0) {
          shape = PartialTensorShape(absl::Span<const int64_t>(
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat May 31 07:13:41 GMT 2025
    - 15.7K bytes
    - Click Count (0)
  4. tensorflow/c/eager/gradients.cc

                             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,
                              const int64_t* dims, const int num_dims,
                              ForwardOperation* forward_op_) {
      if (num_dims > TensorShape::MaxDimensions()) {
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Tue Feb 24 06:18:31 GMT 2026
    - 19.6K bytes
    - Click Count (0)
  5. tensorflow/c/eager/c_api.cc

      }
    }
    
    void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name, const int64_t* dims,
                            const int num_dims, TF_Status* out_status) {
      out_status->status =
          tensorflow::unwrap(op)->SetAttrShape(attr_name, dims, num_dims);
    }
    
    void TFE_OpSetAttrFunction(TFE_Op* op, const char* attr_name,
                               const TFE_Op* value) {
      auto s = tensorflow::unwrap(op)->SetAttrFunction(
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Fri Nov 07 05:55:21 GMT 2025
    - 43.9K bytes
    - Click Count (0)
Back to Top