HomeSort by relevance Sort by last modified time
    Searched defs:TF_Tensor (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/c/
c_api_internal.h 53 struct TF_Tensor {
54 ~TF_Tensor();
133 // buffers of a TF_Tensor pinned in device memory.
191 Status TF_TensorToTensor(const TF_Tensor* src, Tensor* dst);
193 TF_Tensor* TF_TensorFromTensor(const Tensor& src, TF_Status* status);
c_api.h 205 // TF_Tensor holds a multi-dimensional array of elements of a single data type.
221 typedef struct TF_Tensor TF_Tensor;
233 TF_CAPI_EXPORT extern TF_Tensor* TF_NewTensor(
247 TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTensor(TF_DataType,
251 // Deletes `tensor` and returns a new TF_Tensor with the same content if
253 TF_CAPI_EXPORT extern TF_Tensor* TF_TensorMaybeMove(TF_Tensor* tensor);
256 TF_CAPI_EXPORT extern void TF_DeleteTensor(TF_Tensor*);
259 TF_CAPI_EXPORT extern TF_DataType TF_TensorType(const TF_Tensor*);
    [all...]
c_api.cc 177 TF_Tensor::~TF_Tensor() { buffer->Unref(); }
179 TF_Tensor* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims,
186 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims,
201 // TF_Tensor than they do in tensorflow::Tensor. So a copy here is a waste
218 TF_Tensor* ret = new TF_Tensor{dtype, TensorShape(dimvec), buf};
227 TF_Tensor* TF_TensorMaybeMove(TF_Tensor* tensor) {
239 void TF_DeleteTensor(TF_Tensor* t) { delete t;
    [all...]

Completed in 490 milliseconds