Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_SetAttrBoolList (0.07 sec)

  1. tensorflow/c/c_api.cc

    }
    
    void TF_SetAttrBool(TF_OperationDescription* desc, const char* attr_name,
                        unsigned char value) {
      desc->node_builder.Attr(attr_name, static_cast<bool>(value));
    }
    
    void TF_SetAttrBoolList(TF_OperationDescription* desc, const char* attr_name,
                            const unsigned char* values, int num_values) {
      std::unique_ptr<bool[]> b(new bool[num_values]);
      for (int i = 0; i < num_values; ++i) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

    }
    
    TEST_F(CApiAttributesTest, BoolList) {
      const unsigned char list[] = {0, 1, 1, 0, 0, 1, 1};
      const size_t list_size = TF_ARRAYSIZE(list);
    
      auto desc = init("list(bool)");
      TF_SetAttrBoolList(desc, "v", list, list_size);
    
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      unsigned char values[list_size];
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top