Home
last modified time | relevance | path

Searched refs:TFE_TensorHandle (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api.h116 typedef struct TFE_TensorHandle TFE_TensorHandle; typedef
118 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t,
120 TF_CAPI_EXPORT extern void TFE_DeleteTensorHandle(TFE_TensorHandle* h);
121 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h);
122 TF_CAPI_EXPORT extern int TFE_TensorHandleNumDims(TFE_TensorHandle* h);
123 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleDim(TFE_TensorHandle* h,
126 TFE_TensorHandle* h);
127 TF_CAPI_EXPORT extern TF_Tensor* TFE_TensorHandleResolve(TFE_TensorHandle* h,
136 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopyToDevice(
137 TFE_TensorHandle* h, TFE_Context* ctx, const char* device_name,
[all …]
Dc_api_test.cc32 TFE_TensorHandle* TestMatrixTensorHandle() { in TestMatrixTensorHandle()
39 TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); in TestMatrixTensorHandle()
46 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp()
63 TFE_TensorHandle* TestAxisTensorHandle() { in TestAxisTensorHandle()
70 TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); in TestAxisTensorHandle()
77 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, in MinOp()
78 TFE_TensorHandle* axis) { in MinOp()
128 TFE_TensorHandle* m = TestMatrixTensorHandle(); in BM_InitOp()
150 TFE_TensorHandle* m = TestMatrixTensorHandle(); in BM_Execute()
152 TFE_TensorHandle* retvals[1]; in BM_Execute()
[all …]
Dc_api.cc144 TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t, TF_Status* status) { in TFE_NewTensorHandle()
148 return new TFE_TensorHandle(tensor, nullptr); in TFE_NewTensorHandle()
151 void TFE_DeleteTensorHandle(TFE_TensorHandle* h) { delete h; } in TFE_DeleteTensorHandle()
153 TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { in TFE_TensorHandleDataType()
157 int TFE_TensorHandleNumDims(TFE_TensorHandle* h) { return h->t.dims(); } in TFE_TensorHandleNumDims()
159 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index) { in TFE_TensorHandleDim()
163 const char* TFE_TensorHandleDeviceName(TFE_TensorHandle* h) { in TFE_TensorHandleDeviceName()
171 TF_Tensor* TFE_TensorHandleResolve(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleResolve()
186 TFE_TensorHandle* TFE_TensorHandleCopyToDevice(TFE_TensorHandle* h, in TFE_TensorHandleCopyToDevice()
205 return new TFE_TensorHandle(h->t, dst_cpu ? nullptr : dstd); in TFE_TensorHandleCopyToDevice()
[all …]
Dc_api_internal.h93 struct TFE_TensorHandle { struct
94 TFE_TensorHandle(const tensorflow::Tensor& t, tensorflow::Device* d) in TFE_TensorHandle() function
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc48 TFE_TensorHandle* NumpyToTensorHandle(PyObject* obj) { in NumpyToTensorHandle()
65 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle, in EagerCast()
84 TFE_TensorHandle* output = nullptr; in EagerCast()
99 TFE_TensorHandle* CopyToDevice(TFE_TensorHandle* handle, PyObject* ctx, in CopyToDevice()
121 TFE_TensorHandle* new_handle = in CopyToDevice()
175 TFE_TensorHandle* handle;
221 tensorflow::make_safe(static_cast<TFE_TensorHandle*>(nullptr)); in EagerTensor_init()
562 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o) { in EagerTensor_Handle()
566 PyObject* EagerTensorFromHandle(TFE_TensorHandle* handle) { in EagerTensorFromHandle()
689 TFE_TensorHandle* handle = t->handle; in TFE_Py_TensorShapeSlice()
[all …]
Dpywrap_tfe.h24 typedef tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 4>
26 typedef tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2>
91 PyObject* EagerTensorFromHandle(TFE_TensorHandle* handle);
94 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o);
Dpywrap_tfe_src.cc988 TFE_TensorHandle* t = EagerTensor_Handle(tensor);
1374 TFE_TensorHandle* input_handle = EagerTensor_Handle(input);
1680 TFE_TensorHandle* input_handle = EagerTensor_Handle(py_input);
1722 tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2> retvals(num_retvals);
/external/tensorflow/tensorflow/python/lib/core/
Dsafe_ptr.h37 void operator()(TFE_TensorHandle* p) const { TFE_DeleteTensorHandle(p); } in operator()
59 std::unique_ptr<TFE_TensorHandle, detail::TFETensorHandleDeleter>;
60 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle);
Dsafe_ptr.cc28 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle) { in make_safe()
Dpy_func.cc82 arg = EagerTensorFromHandle(new TFE_TensorHandle(t, call->device)); in MakeArgTuple()
85 arg = EagerTensorFromHandle(new TFE_TensorHandle(t, nullptr)); in MakeArgTuple()