Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 10 of 21 for dltensor (0.86 seconds)

The search processing time has exceeded the limit. The displayed results may be partial.

  1. tensorflow/c/eager/dlpack.h

    #ifndef TENSORFLOW_C_EAGER_DLPACK_H_
    #define TENSORFLOW_C_EAGER_DLPACK_H_
    
    #include "tensorflow/c/eager/c_api.h"
    
    namespace tensorflow {
    
    // PyCapsule name for DLPack Tensor
    const char* const kDlTensorCapsuleName = "dltensor";
    
    // Returns the DLDevice* for the given eager tensor handle.
    //
    // The caller takes ownership of the returned pointer and is responsible for
    // deleting it.
    TF_CAPI_EXPORT extern void* TFE_GetDLDevice(TFE_TensorHandle* h,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Mar 13 23:41:52 GMT 2025
    - 1.9K bytes
    - Click Count (0)
  2. tensorflow/c/eager/dlpack_test.cc

        num_elements *= shape[i];
      }
      std::vector<float> data(num_elements);
      for (size_t j = 0; j < num_elements; ++j) {
        data[j] = j;
      }
      DLManagedTensor dlm_in = {};
      DLTensor* dltensor_in = &dlm_in.dl_tensor;
      dltensor_in->data = data.data();
      dltensor_in->device = {kDLCPU, 0};
      dltensor_in->ndim = static_cast<int32_t>(shape.size());
      dltensor_in->dtype = {kDLFloat, 32, 1};
    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)
  3. tensorflow/c/eager/c_api_debug.cc

        TFE_TensorHandle* h, TF_Status* status) {
      tensorflow::TensorHandle* handle =
          TensorHandleFromInterface(tensorflow::unwrap(h));
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return nullptr;
      }
    
      std::vector<int64_t> dev_dims = TensorShapeAsVector(*handle, &status->status);
      if (!status->status.ok()) {
        return nullptr;
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 2.5K bytes
    - Click Count (0)
  4. tensorflow/c/eager/immediate_execution_tensor_handle.h

      // Returns a tensor for the handle. If tensor is remote, it will be copied.
      virtual AbstractTensorInterface* Resolve(absl::Status* status) = 0;
    
      std::string DebugString() const override;
    
      // Returns a Boolean hint indicating whether callers should prefer
      // `SummarizeValue` to resolving this handle and formatting the tensor.
      //
      // For example some tensor handles may represent distributed values, in which
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 4.3K bytes
    - Click Count (0)
  5. tensorflow/c/eager/c_api_test_util.h

    #include "tensorflow/core/protobuf/tensorflow_server.pb.h"
    
    // Return a tensor handle containing a float scalar
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, float value);
    
    // Return a tensor handle containing a int scalar
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, int value);
    
    // Return a tensor handle containing a bool scalar
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, bool value);
    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)
  6. tensorflow/c/eager/custom_device_testutil.cc

    };
    
    struct LoggedTensor {
      TFE_TensorHandle* tensor;
      LoggedTensor() = delete;
      explicit LoggedTensor(TFE_TensorHandle* tensor) : tensor(tensor) {}
      ~LoggedTensor() { TFE_DeleteTensorHandle(tensor); }
    };
    
    int64_t LoggedTensorDim(void* data, int dim_index, TF_Status* status) {
      return TFE_TensorHandleDim(reinterpret_cast<LoggedTensor*>(data)->tensor,
                                 dim_index, status);
    }
    
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Dec 20 23:30:21 GMT 2025
    - 8.4K bytes
    - Click Count (0)
  7. tensorflow/c/eager/abstract_tensor_handle.h

      // Returns the status of the tensor handle. If it is a tfrt::TensorHandle,
      // the tensor handle can be an error and return non-OK status.
      virtual absl::Status TensorHandleStatus() const;
    
      // Returns tensor shape. If tensor has unknown rank, shape remains untouched.
      virtual absl::Status Shape(tensorflow::PartialTensorShape* shape) const = 0;
    
      // Returns tensor (full) type.
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 3K bytes
    - Click Count (0)
  8. 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)
  9. tensorflow/c/checkpoint_reader.h

      // tensor are combined into a single entry.
      const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;
    
      // Returns a map from variable names to their data types.  Slices of a
      // partitioned tensor are combined into a single entry.
      const TensorSliceReader::VarToDataTypeMap& GetVariableToDataTypeMap() const;
    
      // Attempts to look up the tensor named "name" and stores the found result in
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Jan 07 04:56:09 GMT 2026
    - 3.1K bytes
    - Click Count (0)
  10. tensorflow/c/eager/gradient_checker.cc

        (*data)[i] = i;
      }
    }
    
    // Fills out_dims with the dimensions of the given tensor.
    void GetDims(const TF_Tensor* t, int64_t* out_dims) {
      int num_dims = TF_NumDims(t);
      for (int i = 0; i < num_dims; i++) {
        out_dims[i] = TF_Dim(t, i);
      }
    }
    
    // Runs model as is if output is a scalar,
    // else sums the output tensor before returning.
    absl::Status RunAndMaybeSum(AbstractContext* ctx, Model forward,
    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)
Back to Top