Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 5 of 5 for ndims (0.03 seconds)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to Top