Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 11 for SetAttrShape (0.28 sec)

  1. tensorflow/c/eager/abstract_operation.h

      virtual Status SetAttrBool(const char* attr_name, bool value) = 0;
      virtual Status SetAttrType(const char* attr_name, DataType value) = 0;
      virtual Status SetAttrShape(const char* attr_name, const int64_t* dims,
                                  const int num_dims) = 0;
      virtual Status SetAttrShape(const char* attr_name,
                                  const PartialTensorShape shape);
      virtual Status SetAttrFunction(const char* attr_name,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/gradients/tape/tape_operation.h

      Status SetAttrFloat(const char* attr_name, float value) override;
      Status SetAttrBool(const char* attr_name, bool value) override;
      Status SetAttrType(const char* attr_name, DataType value) override;
      Status SetAttrShape(const char* attr_name, const int64_t* dims,
                          const int num_dims) override;
      Status SetAttrFunction(const char* attr_name,
                             const AbstractOperation* value) override;
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 23 23:12:39 GMT 2020
    - 3.7K bytes
    - Viewed (1)
  3. tensorflow/c/experimental/gradients/tape/tape_operation.cc

    }
    Status TapeOperation::SetAttrType(const char* attr_name, DataType value) {
      forward_op_.attrs.Set(attr_name, value);
      return parent_op_->SetAttrType(attr_name, value);
    }
    Status TapeOperation::SetAttrShape(const char* attr_name, const int64_t* dims,
                                       const int num_dims) {
      if (num_dims > TensorShape::MaxDimensions()) {
        return errors::InvalidArgument("Value specified for `", attr_name, "` has ",
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  4. tensorflow/c/eager/gradients_internal.h

                       ForwardOperation*);
    Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value,
                       ForwardOperation*);
    Status SetAttrShape(AbstractOperation*, const char* attr_name,
                        const int64_t* dims, const int num_dims, ForwardOperation*);
    Status SetAttrFunction(AbstractOperation*, const char* attr_name,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:35 GMT 2021
    - 4.2K bytes
    - Viewed (0)
  5. 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();
      }
      Status SetAttrShape(const char* attr_name, const int64_t* dims,
                          const int num_dims) override {
        PartialTensorShape shape;
        if (num_dims >= 0) {
          shape = PartialTensorShape(ArraySlice<int64_t>(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  6. tensorflow/c/eager/gradients.cc

                       DataType value, ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrType(attr_name, value);
    }
    Status SetAttrShape(AbstractOperation* op_, const char* attr_name,
                        const int64_t* dims, const int num_dims,
                        ForwardOperation* forward_op_) {
      if (num_dims > TensorShape::MaxDimensions()) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/c/c_test_util.cc

                           TF_Operation** op) {
      TF_OperationDescription* desc = TF_NewOperation(graph, "Placeholder", name);
      TF_SetAttrType(desc, "dtype", dtype);
      if (!dims.empty()) {
        TF_SetAttrShape(desc, "shape", dims.data(), dims.size());
      }
      *op = TF_FinishOperation(desc, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      ASSERT_NE(*op, nullptr);
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  8. 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(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  9. tensorflow/c/c_api_test.cc

      EXPECT_TRUE(std::equal(std::begin(list), std::end(list), std::begin(values)));
    }
    
    TEST_F(CApiAttributesTest, Shape) {
      // Unknown shape
      auto desc = init("shape");
      TF_SetAttrShape(desc, "v", nullptr, -1);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  10. tensorflow/c/c_api.h

    // Set `num_dims` to -1 to represent "unknown rank".  Otherwise,
    // `dims` points to an array of length `num_dims`.  `dims[i]` must be
    // >= -1, with -1 meaning "unknown dimension".
    TF_CAPI_EXPORT extern void TF_SetAttrShape(TF_OperationDescription* desc,
                                               const char* attr_name,
                                               const int64_t* dims, int num_dims);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
Back to top