- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TF_SetAttrInt (0.05 sec)
-
tensorflow/c/c_test_util.cc
ScalarConst(0, graph, s, absl::StrCat(name, "_const0").c_str()); TF_OperationDescription* desc = TF_NewOperation(graph, "Split", name); TF_AddInput(desc, {zero, 0}); TF_AddInput(desc, {input, 0}); TF_SetAttrInt(desc, "num_split", 3); TF_SetAttrType(desc, "T", TF_INT32); // Set device to CPU since there is no version of split for int32 on GPU // TODO(iga): Convert all these helpers and tests to use floats because
Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Sat Oct 04 05:55:32 UTC 2025 - 17.8K bytes - Viewed (1) -
tensorflow/c/c_api_test.cc
EXPECT_EQ(list[i], string(static_cast<const char*>(values[i]), lens[i])) << i; } } TEST_F(CApiAttributesTest, Int) { auto desc = init("int"); TF_SetAttrInt(desc, "v", 31415); auto oper = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); EXPECT_TF_META("v", -1, TF_ATTR_INT, -1); int64_t value;
Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Mon Nov 17 00:00:38 UTC 2025 - 97K bytes - Viewed (0) -
tensorflow/c/c_api.h
const size_t* lengths, int num_values); 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,Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
tensorflow/c/c_api.cc
v.reserve(num_values); for (int i = 0; i < num_values; ++i) { v.emplace_back(static_cast<const char*>(values[i]), lengths[i]); } desc->node_builder.Attr(attr_name, v); } } void TF_SetAttrInt(TF_OperationDescription* desc, const char* attr_name, int64_t value) { desc->node_builder.Attr(attr_name, static_cast<int64_t>(value)); }
Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Sat Oct 04 05:55:32 UTC 2025 - 102.4K bytes - Viewed (0)