- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for SetAttrIntList (0.23 sec)
-
tensorflow/c/eager/gradients.cc
} Status SetAttrIntList(AbstractOperation* op_, const char* attr_name, const int64_t* values, int num_values, ForwardOperation* forward_op_) { forward_op_->attrs.Set( attr_name, gtl::ArraySlice<const int64_t>( reinterpret_cast<const int64_t*>(values), num_values)); return op_->SetAttrIntList(attr_name, values, num_values); }
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/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>(
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) -
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;
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) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
int num_values) override { op_->node_builder.Attr(attr_name, ArraySlice<const float>(values, num_values)); return absl::OkStatus(); } Status SetAttrIntList(const char* attr_name, const int64_t* values, int num_values) override { op_->node_builder.Attr( attr_name, ArraySlice<const int64_t>(
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
} } void TFE_OpSetAttrIntList(TFE_Op* op, const char* attr_name, const int64_t* values, int num_values) { auto s = tensorflow::unwrap(op)->SetAttrIntList(attr_name, values, num_values); if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } void TFE_OpSetAttrTypeList(TFE_Op* op, const char* attr_name,
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(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];
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)