- Sort Score
- Num 10 results
- Language All
Results 1 - 5 of 5 for ndims (0.03 seconds)
-
tensorflow/c/eager/dlpack_test.cc
EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id); EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim); EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code); EXPECT_EQ(dltensor_out->dtype.bits, dltensor_in->dtype.bits); EXPECT_EQ(dltensor_out->dtype.lanes, dltensor_in->dtype.lanes); for (int i = 0; i < dltensor_in->ndim; ++i) { EXPECT_EQ(dltensor_out->shape[i], dltensor_in->shape[i]); if (dltensor_out->strides) {
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Click Count (0) -
tensorflow/c/eager/c_api_test_util.h
float data[], int64_t dims[], int num_dims); // Get a Matrix TensorHandle with given float values and dimensions TFE_TensorHandle* TestTensorHandleWithDimsFloat(TFE_Context* ctx, float data[], int64_t dims[], int num_dims); // Get a Matrix TensorHandle with given int values and dimensions
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Click Count (0) -
tensorflow/c/c_test_util.h
TF_Tensor* BoolTensor(int32_t v); // Create a tensor with values of type TF_INT8 provided by `values`. TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values); // Create a tensor with values of type TF_INT32 provided by `values`. TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims, const int32_t* values); // Create 1 dimensional tensor with values from `values`
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Aug 09 01:06:53 GMT 2018 - 6K bytes - Click Count (0) -
tensorflow/c/eager/abstract_operation.h
virtual absl::Status SetAttrBool(const char* attr_name, bool value) = 0; virtual absl::Status SetAttrType(const char* attr_name, DataType value) = 0; virtual absl::Status SetAttrShape(const char* attr_name, const int64_t* dims, const int num_dims) = 0; virtual absl::Status SetAttrShape(const char* attr_name, const PartialTensorShape shape);
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Oct 12 05:11:17 GMT 2024 - 7.3K bytes - Click Count (0) -
tensorflow/c/eager/gradients_internal.h
absl::Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value, ForwardOperation*); absl::Status SetAttrShape(AbstractOperation*, const char* attr_name, const int64_t* dims, const int num_dims, ForwardOperation*); absl::Status SetAttrFunction(AbstractOperation*, const char* attr_name, const AbstractOperation* value, ForwardOperation*);
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Oct 12 05:11:17 GMT 2024 - 4.7K bytes - Click Count (0)