Home
last modified time | relevance | path

Searched refs:TFE_AllocateHostTensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc37 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status); in TestScalarTensorHandle()
49 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_STRING, nullptr, 0, status); in TestScalarTensorHandle()
62 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_INT32, nullptr, 0, status); in TestScalarTensorHandle()
74 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_BOOL, nullptr, 0, status); in TestScalarTensorHandle()
87 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_DOUBLE, &dims[0], in DoubleTestMatrixTensorHandle()
101 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in TestMatrixTensorHandle()
116 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status); in TestMatrixTensorHandleWithInput()
129 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status); in TestTensorHandleWithDimsFloat()
142 TFE_AllocateHostTensor(ctx, TF_INT32, &dims[0], num_dims, status); in TestTensorHandleWithDimsInt()
159 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in TestMatrixTensorHandle100x100()
[all …]
Dc_api_test_util.h64 TF_Tensor* t = TFE_AllocateHostTensor(ctx, datatype, dims, num_dims, status); in TestTensorHandleWithDims()
Dc_api_experimental.h603 TF_CAPI_EXPORT extern TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx,
Dc_api_experimental.cc632 TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype, in TFE_AllocateHostTensor() function