Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_SetAttrIntList (0.15 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_test.cc

                       merge_op.length());
      const std::string final_op("Id");
      TF_SetAttrString(reduce_desc, "final_op", final_op.c_str(),
                       final_op.length());
      TF_SetAttrIntList(reduce_desc, "subdiv_offsets", nullptr, 0);
      TF_AddInput(reduce_desc, x);
      TF_Operation* reduce_op = TF_FinishOperation(reduce_desc, status);
      if (TF_GetCode(status) != TF_OK) return;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

      EXPECT_EQ(31415, value);
    }
    
    TEST_F(CApiAttributesTest, IntList) {
      const int64_t list[] = {1, 2, 3, 4};
      const size_t list_size = TF_ARRAYSIZE(list);
    
      auto desc = init("list(int)");
      TF_SetAttrIntList(desc, "v", list, list_size);
    
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      int64_t 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)
  3. tensorflow/c/c_api.h

    TF_CAPI_EXPORT extern void TF_SetAttrInt(TF_OperationDescription* desc,
                                             const char* attr_name, int64_t value);
    TF_CAPI_EXPORT extern void TF_SetAttrIntList(TF_OperationDescription* desc,
                                                 const char* attr_name,
                                                 const int64_t* values,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api.cc

      }
    }
    
    void TF_SetAttrInt(TF_OperationDescription* desc, const char* attr_name,
                       int64_t value) {
      desc->node_builder.Attr(attr_name, static_cast<int64_t>(value));
    }
    
    void TF_SetAttrIntList(TF_OperationDescription* desc, const char* attr_name,
                           const int64_t* values, int num_values) {
      desc->node_builder.Attr(
          attr_name, ArraySlice<const int64_t>(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top