Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_SetAttrShape (0.08 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);
    }
    
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Sat Oct 04 05:55:32 UTC 2025
    - 17.8K bytes
    - Viewed (1)
  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);
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Mon Nov 17 00:00:38 UTC 2025
    - 97K bytes
    - Viewed (0)
  3. 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(absl::Span<const int64_t>(
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Sat Oct 04 05:55:32 UTC 2025
    - 102.4K bytes
    - Viewed (0)
Back to top