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
57
struct
TF_Tensor
{
58
~
TF_Tensor
();
206
Status TF_TensorToTensor(const
TF_Tensor
* src, Tensor* dst);
208
TF_Tensor
* TF_TensorFromTensor(const Tensor& src, TF_Status* status);
c_api.h
206
//
TF_Tensor
holds a multi-dimensional array of elements of a single data type.
222
typedef struct
TF_Tensor
TF_Tensor
;
234
TF_CAPI_EXPORT extern
TF_Tensor
* TF_NewTensor(
248
TF_CAPI_EXPORT extern
TF_Tensor
* TF_AllocateTensor(TF_DataType,
252
// Deletes `tensor` and returns a new
TF_Tensor
with the same content if
254
TF_CAPI_EXPORT extern
TF_Tensor
* TF_TensorMaybeMove(
TF_Tensor
* tensor);
257
TF_CAPI_EXPORT extern void TF_DeleteTensor(
TF_Tensor
*);
260
TF_CAPI_EXPORT extern TF_DataType TF_TensorType(const
TF_Tensor
*);
[
all
...]
c_api.cc
195
TF_Tensor
::~
TF_Tensor
() { buffer->Unref(); }
197
TF_Tensor
* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims,
204
TF_Tensor
* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims,
219
//
TF_Tensor
than they do in tensorflow::Tensor. So a copy here is a waste
234
TF_Tensor
* ret = new
TF_Tensor
{dtype, TensorShape(dimvec), buf};
243
TF_Tensor
* TF_TensorMaybeMove(
TF_Tensor
* tensor) {
255
void TF_DeleteTensor(
TF_Tensor
* t) { delete t;
[
all
...]
Completed in 186 milliseconds