Searched refs:TFE_AllocateHostTensor (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/c/eager/ |
D | c_api_test_util.cc | 34 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 …]
|
D | gradients_util.cc | 46 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()
|
D | c_api_experimental.h | 552 TF_CAPI_EXPORT extern TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx,
|
D | c_api_experimental.cc | 576 TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype, in TFE_AllocateHostTensor() function
|