- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_SetAttrInt (0.21 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TF_OperationDescription* reduce_desc = TF_NewOperation(body.get(), "CollectiveReduce", "CollectiveReduce"); TF_SetAttrType(reduce_desc, "T", TF_FLOAT); TF_SetAttrInt(reduce_desc, "group_size", group_size); 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(),
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Aug 06 23:56:17 UTC 2024 - 29.4K bytes - Viewed (0) -
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 Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 97K 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 Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 102.3K bytes - Viewed (0)