- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_OperationGetAttrShapeList (0.22 sec)
-
tensorflow/c/c_api.h
// TF_AttrMetadata.total_size from TF_OperationGetAttrMetadata(oper, // attr_name). // // Fails if storage_size is insufficient to hold the requested shapes. TF_CAPI_EXPORT extern void TF_OperationGetAttrShapeList( TF_Operation* oper, const char* attr_name, int64_t** dims, int* num_dims, int num_shapes, int64_t* storage, int storage_size, TF_Status* status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
tensorflow/c/c_api.cc
if (!status->status.ok()) return; auto len = std::min(shape.dims(), num_dims); for (int i = 0; i < len; ++i) { value[i] = shape.dim_size(i); } } void TF_OperationGetAttrShapeList(TF_Operation* oper, const char* attr_name, int64_t** dims, int* num_dims, int num_shapes, int64_t* storage, int storage_size,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
EXPECT_TF_META("v", list_size, TF_ATTR_SHAPE, total_ndims); int64_t* values[list_size]; int values_ndims[list_size]; int64_t storage[total_ndims]; TF_OperationGetAttrShapeList(oper, "v", values, values_ndims, list_size, storage, total_ndims, s_); EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); for (size_t i = 0; i < list_size; ++i) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 97K bytes - Viewed (0)