- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for SetAttrInt (0.22 sec)
-
tensorflow/c/eager/gradients.cc
return op_->SetAttrString(attr_name, data, length); } Status SetAttrInt(AbstractOperation* op_, const char* attr_name, int64_t value, ForwardOperation* forward_op_) { forward_op_->attrs.Set(attr_name, static_cast<int64_t>(value)); return op_->SetAttrInt(attr_name, value); } Status SetAttrFloat(AbstractOperation* op_, const char* attr_name, float value,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
size_t length) override { tensorflow::StringPiece s(data, length); op_->node_builder.Attr(attr_name, s); return absl::OkStatus(); } Status SetAttrInt(const char* attr_name, int64_t value) override { op_->node_builder.Attr(attr_name, static_cast<int64_t>(value)); return absl::OkStatus(); } Status SetAttrFloat(const char* attr_name, float value) override {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api.cc
if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name, int64_t value) { auto s = tensorflow::unwrap(op)->SetAttrInt(attr_name, value); if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } void TFE_OpSetAttrFloat(TFE_Op* op, const char* attr_name, float value) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
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;
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) -
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,
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) -
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)); }
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)