Searched refs:TFE_TensorHandle (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/c/eager/ |
D | c_api.h | 116 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 …]
|
D | c_api_test.cc | 32 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 …]
|
D | c_api.cc | 144 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 …]
|
D | c_api_internal.h | 93 struct TFE_TensorHandle { struct 94 TFE_TensorHandle(const tensorflow::Tensor& t, tensorflow::Device* d) in TFE_TensorHandle() function
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tensor.cc | 48 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 …]
|
D | pywrap_tfe.h | 24 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);
|
D | pywrap_tfe_src.cc | 988 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/ |
D | safe_ptr.h | 37 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);
|
D | safe_ptr.cc | 28 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle) { in make_safe()
|
D | py_func.cc | 82 arg = EagerTensorFromHandle(new TFE_TensorHandle(t, call->device)); in MakeArgTuple() 85 arg = EagerTensorFromHandle(new TFE_TensorHandle(t, nullptr)); in MakeArgTuple()
|