Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for B_dims (0.08 sec)

  1. tensorflow/c/eager/gradient_checker_test.cc

      int64_t A_dims[] = {2, 2};
      AbstractTensorHandlePtr A;
      {
        AbstractTensorHandle* A_raw;
        absl::Status s = TestTensorHandleWithDims<float, TF_FLOAT>(
            ctx_.get(), A_vals, A_dims, 2, &A_raw);
        ASSERT_EQ(errors::OK, s.code()) << s.message();
        A.reset(A_raw);
      }
      float B_vals[] = {.5f, -1.0f, 1.0f, 1.0f};
      int64_t B_dims[] = {2, 2};
      AbstractTensorHandlePtr B;
      {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.5K bytes
    - Viewed (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
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Jul 17 23:43:59 UTC 2023
    - 7.7K bytes
    - Viewed (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`
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 09 01:06:53 UTC 2018
    - 6K bytes
    - Viewed (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);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (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*);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  6. tensorflow/c/eager/dlpack_test.cc

      TestHandleFromDLPack(status, ctx, {4}, {});
      TestHandleFromDLPack(status, ctx, {4}, {1});
      TestHandleFromDLPack(status, ctx, {4, 3, 2}, {});
      TestHandleFromDLPack(status, ctx, {4, 3, 2}, {6, 2, 1});
      // Test that dims with size=1 can have any stride.
      TestHandleFromDLPack(status, ctx, {1}, {1});
      TestHandleFromDLPack(status, ctx, {1}, {0});
      TestHandleFromDLPack(status, ctx, {4, 1, 2}, {2, 1, 1});
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Jun 30 03:04:46 UTC 2023
    - 4.4K bytes
    - Viewed (0)
Back to top