Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_SetAttrShape (0.26 sec)

  1. 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)
  2. 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)
  3. 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)
  4. tensorflow/c/c_api.cc

                            const char* value, size_t length) {
      tensorflow::NameAttrList func_name;
      func_name.set_name(string(value, value + length));
      desc->node_builder.Attr(attr_name, func_name);
    }
    
    void TF_SetAttrShape(TF_OperationDescription* desc, const char* attr_name,
                         const int64_t* dims, int num_dims) {
      PartialTensorShape shape;
      if (num_dims >= 0) {
        shape = PartialTensorShape(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top