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.cc34 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status); in TestScalarTensorHandle()
46 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_STRING, nullptr, 0, status); in TestScalarTensorHandle()
59 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_INT32, nullptr, 0, status); in TestScalarTensorHandle()
71 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_BOOL, nullptr, 0, status); in TestScalarTensorHandle()
84 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_DOUBLE, &dims[0], in DoubleTestMatrixTensorHandle()
98 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in TestMatrixTensorHandle()
113 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status); in TestMatrixTensorHandleWithInput()
126 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status); in TestTensorHandleWithDimsFloat()
139 TFE_AllocateHostTensor(ctx, TF_INT32, &dims[0], num_dims, status); in TestTensorHandleWithDimsInt()
156 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in TestMatrixTensorHandle100x100()
[all …]
Dgradients_util.cc46 TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status.get()); in ScalarTensorHandleHelper()
60 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status.get()); in TensorHandleWithDimsFloatHelper()
74 TFE_AllocateHostTensor(ctx, TF_INT32, &dims[0], num_dims, status.get()); in TensorHandleWithDimsIntHelper()
Dc_api_experimental.h552 TF_CAPI_EXPORT extern TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx,
Dc_api_experimental.cc576 TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype, in TFE_AllocateHostTensor() function