- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TFE_OpSetAttrIntList (0.1 sec)
-
tensorflow/c/eager/c_api.h
const size_t* lengths, int num_values); TF_CAPI_EXPORT extern void TFE_OpSetAttrIntList(TFE_Op* op, const char* attr_name, const int64_t* values, int num_values);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
TFE_OpSetAttrInt(op, "instance_key", 456); TFE_OpSetAttrString(op, "merge_op", "Add", 3); TFE_OpSetAttrString(op, "final_op", "Id", 2); std::vector<int64_t> subdiv_offsets; TFE_OpSetAttrIntList(op, "subdiv_offsets", subdiv_offsets.data(), subdiv_offsets.size()); return op; } TFE_Op* SendOp(TFE_Context* ctx, TFE_TensorHandle* in,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
auto s = tensorflow::unwrap(op)->SetAttrFloatList(attr_name, values, num_values); if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } 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()) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
merge_op.length()); const std::string final_op("Id"); TFE_OpSetAttrString(op.get(), "final_op", final_op.c_str(), final_op.length()); TFE_OpSetAttrIntList(op.get(), "subdiv_offsets", nullptr, 0); TFE_OpAddInput(op.get(), input, status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_TensorHandle* result_handle; int num_retvals = 1;
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/eager/c_api_test.cc
TFE_OpAddInput(op, tensor_handle, status); TF_DeleteTensor(tensor); TFE_DeleteTensorHandle(tensor_handle); std::vector<int64_t> values(4, 1); TFE_OpSetAttrIntList(op, "ksize", values.data(), values.size()); TFE_OpSetAttrIntList(op, "strides", values.data(), values.size()); const int BUFFER_SIZE = 10; char buffer[BUFFER_SIZE]; std::strncpy(buffer, "VALID", BUFFER_SIZE);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0)