Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TF_SetAttrString (0.52 sec)

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

      TF_SetAttrInt(reduce_desc, "group_key", 0);
      TF_SetAttrInt(reduce_desc, "instance_key", 0);
    
      const std::string merge_op("Mul");
      TF_SetAttrString(reduce_desc, "merge_op", merge_op.c_str(),
                       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);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  2. tensorflow/c/c_api_test.cc

        EXPECT_EQ(expected_total_size, m.total_size);                    \
      } while (0)
    
    TEST_F(CApiAttributesTest, String) {
      auto desc = init("string");
      TF_SetAttrString(desc, "v", "bunny", 5);
    
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", -1, TF_ATTR_STRING, 5);
    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

    // inferred from an input and doesn't have a default value you wish to
    // keep.
    
    // `value` must point to a string of length `length` bytes.
    TF_CAPI_EXPORT extern void TF_SetAttrString(TF_OperationDescription* desc,
                                                const char* attr_name,
                                                const void* value, size_t length);
    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_function_test.cc

                            TF_Operation** op) {
      TF_OperationDescription* desc = TF_NewOperation(graph, "Placeholder", name);
      TF_SetAttrType(desc, "dtype", TF_INT32);
      TF_SetAttrString(desc, attr_name, attr_value, strlen(attr_value));
      *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: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  5. tensorflow/c/c_api.cc

    }
    
    void TF_ColocateWith(TF_OperationDescription* desc, TF_Operation* op) {
      desc->colocation_constraints.emplace(
          StrCat(tensorflow::kColocationGroupPrefix, op->node.name()));
    }
    
    void TF_SetAttrString(TF_OperationDescription* desc, const char* attr_name,
                          const void* value, size_t length) {
      tensorflow::StringPiece s(static_cast<const char*>(value), length);
      desc->node_builder.Attr(attr_name, s);
    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