OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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