Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_SetAttrShape (0.15 sec)

  1. 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)
  2. 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