Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.h22 TFE_TensorHandle* TestScalarTensorHandle(float value);
25 TFE_TensorHandle* TestScalarTensorHandle(int value);
28 TFE_TensorHandle* TestScalarTensorHandle(bool value);
31 TFE_TensorHandle* DoubleTestMatrixTensorHandle();
34 TFE_TensorHandle* TestMatrixTensorHandle();
37 TFE_TensorHandle* TestMatrixTensorHandle100x100();
40 TFE_TensorHandle* DoubleTestMatrixTensorHandle3X2();
43 TFE_TensorHandle* TestMatrixTensorHandle3X2();
46 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
49 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a);
[all …]
Dc_api_test_util.cc24 TFE_TensorHandle* TestScalarTensorHandle(float value) { in TestScalarTensorHandle()
29 TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); in TestScalarTensorHandle()
36 TFE_TensorHandle* TestScalarTensorHandle(int value) { in TestScalarTensorHandle()
41 TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); in TestScalarTensorHandle()
48 TFE_TensorHandle* TestScalarTensorHandle(bool value) { in TestScalarTensorHandle()
53 TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); in TestScalarTensorHandle()
60 TFE_TensorHandle* DoubleTestMatrixTensorHandle() { in DoubleTestMatrixTensorHandle()
67 TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); in DoubleTestMatrixTensorHandle()
74 TFE_TensorHandle* TestMatrixTensorHandle() { in TestMatrixTensorHandle()
81 TFE_TensorHandle* th = TFE_NewTensorHandle(t, status); in TestMatrixTensorHandle()
[all …]
Dc_api.h138 typedef struct TFE_TensorHandle TFE_TensorHandle; typedef
140 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t,
143 TF_CAPI_EXPORT extern void TFE_DeleteTensorHandle(TFE_TensorHandle* h);
144 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h);
146 TF_CAPI_EXPORT extern int TFE_TensorHandleNumDims(TFE_TensorHandle* h,
148 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h,
151 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleDim(TFE_TensorHandle* h,
161 TFE_TensorHandle* h, TF_Status* status);
167 TFE_TensorHandle* h, TF_Status* status);
172 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor(
[all …]
Dc_api_test.cc47 TFE_TensorHandle* m = TestMatrixTensorHandle(); in BM_InitOp()
71 TFE_TensorHandle* m = TestMatrixTensorHandle(); in BM_Execute()
73 TFE_TensorHandle* retvals[1]; in BM_Execute()
106 TFE_TensorHandle* m = TestMatrixTensorHandle(); in BM_Execute_Identity()
108 TFE_TensorHandle* retvals[1]; in BM_Execute_Identity()
153 TFE_TensorHandle* h = TestMatrixTensorHandle(); in TEST()
179 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(); in TensorHandleCopyBetweenDevices()
196 TFE_TensorHandle* hdevice = in TensorHandleCopyBetweenDevices()
203 TFE_TensorHandle* hdevice2 = in TensorHandleCopyBetweenDevices()
211 TFE_TensorHandle* hcopy = in TensorHandleCopyBetweenDevices()
[all …]
Dc_api_remote_test.cc76 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(); in TestRemoteExecute()
77 TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(); in TestRemoteExecute()
91 TFE_TensorHandle* retvals[1]; in TestRemoteExecute()
161 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(); in TestRemoteExecuteSilentCopies()
162 TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(); in TestRemoteExecuteSilentCopies()
175 TFE_TensorHandle* retvals[1]; in TestRemoteExecuteSilentCopies()
249 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle100x100(); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
250 TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle100x100(); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
264 TFE_TensorHandle* retvals[1]; in TestRemoteExecuteDeleteContextWithOutstandingRPC()
292 void CheckTFE_TensorHandleHasFloats(TFE_TensorHandle* handle, in CheckTFE_TensorHandleHasFloats()
[all …]
Dc_api_experimental_test.cc51 TFE_TensorHandle* m = TestMatrixTensorHandle(); in ExecuteWithProfiling()
53 TFE_TensorHandle* retvals[1] = {nullptr}; in ExecuteWithProfiling()
337 TFE_TensorHandle* h = TFE_NewTensorHandle(t, status); in TEST()
346 std::vector<TFE_TensorHandle*> result; in TEST()
413 TFE_TensorHandle* h = TFE_NewTensorHandle(t, status); in TEST()
425 std::vector<TFE_TensorHandle*> result; in TEST()
464 TFE_TensorHandle* m = TestMatrixTensorHandle(); in Executor_MatMul_CPU()
466 TFE_TensorHandle* retvals[2] = {nullptr, nullptr}; in Executor_MatMul_CPU()
499 TFE_DeleteTensorHandle(static_cast<TFE_TensorHandle*>(tensor_handle)); in Deleter()
509 TFE_TensorHandle* m = TestMatrixTensorHandle(); in TEST()
[all …]
Dc_api.cc896 TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t, TF_Status* status) { in TFE_NewTensorHandle()
900 return TFE_TensorHandle::CreateLocalHandle(tensor, status); in TFE_NewTensorHandle()
903 void TFE_DeleteTensorHandle(TFE_TensorHandle* h) { in TFE_DeleteTensorHandle()
928 TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { in TFE_TensorHandleDataType()
936 int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleNumDims()
956 int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleNumElements()
976 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, in TFE_TensorHandleDim()
998 const char* TFE_TensorHandleDeviceName(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleDeviceName()
1017 const char* TFE_TensorHandleBackingDeviceName(TFE_TensorHandle* h, in TFE_TensorHandleBackingDeviceName()
1037 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor( in TFE_TensorHandleCopySharingTensor()
[all …]
Dc_api_debug_test.cc24 TFE_TensorHandle* h = TestScalarTensorHandle(1.0f); in TEST()
37 TFE_TensorHandle* h = TestMatrixTensorHandle3X2();
Dc_api_experimental.h37 TF_CAPI_EXPORT extern void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h,
443 TF_CAPI_EXPORT extern void* TFE_TensorHandleDevicePointer(TFE_TensorHandle*,
449 TF_CAPI_EXPORT extern size_t TFE_TensorHandleDeviceMemorySize(TFE_TensorHandle*,
455 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromDeviceMemory(
Dc_api_internal.h69 struct TFE_TensorHandle { struct
70 static TFE_TensorHandle* CreateLocalHandle(const class tensorflow::Tensor& t, in CreateLocalHandle() argument
77 return new TFE_TensorHandle{ in CreateLocalHandle()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.h166 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_DequeueNamedTensor(
170 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(
175 TFE_TensorHandle* tensor,
179 TFE_Context* ctx, int tensor_id, TFE_TensorHandle* tensor,
185 TFE_TensorHandle* tensor,
188 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_DequeueVariantTensor(
193 TFE_TensorHandle* handle);
208 TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals,
288 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromScalar(
Dc_api_experimental.cc349 static TFE_TensorHandle* createTFEQueue(TFE_Context* ctx, TF_DataType inputType, in createTFEQueue()
371 TFE_TensorHandle* queue = nullptr; in createTFEQueue()
380 TFE_TensorHandle* queue, TFE_TensorHandle* tensor, in createTFEEnqueue()
400 static TFE_TensorHandle* createTFEDequeue(TFE_Context* ctx, in createTFEDequeue()
402 TFE_TensorHandle* queue, in createTFEDequeue()
414 TFE_TensorHandle* ret; in createTFEDequeue()
422 TFE_TensorHandle* TFE_DequeueNamedTensor(TF_Session* session, int tensor_id, in TFE_DequeueNamedTensor()
433 TFE_TensorHandle* queue = createTFEQueue(ctx, inputType, tensor_id, status); in TFE_DequeueNamedTensor()
435 std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> in TFE_DequeueNamedTensor()
442 TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_DequeueNamedTensorFromCtx()
[all …]
Dc_api_experimental_test.cc94 TFE_TensorHandle* m = TestMatrixTensorHandle(); in TEST()
98 TFE_TensorHandle* retvals[1] = {nullptr}; in TEST()
137 TFE_TensorHandle* m = TestMatrixTensorHandle(); in TEST()
172 TFE_TensorHandle* send_retvals; in TEST()
177 TFE_TensorHandle* recv_retvals[1] = {nullptr}; in TEST()
342 TFE_TensorHandle* scalar = TestScalarTensorHandle(five); in TEST_F()
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_jni.cc39 TFE_TensorHandle* requireTensorHandle(JNIEnv* env, jlong handle) { in requireTensorHandle()
44 return reinterpret_cast<TFE_TensorHandle*>(handle); in requireTensorHandle()
59 TFE_DeleteTensorHandle(reinterpret_cast<TFE_TensorHandle*>(handle)); in Java_org_tensorflow_EagerOperation_deleteTensorHandle()
64 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_resolveTensorHandle()
112 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_dataType()
120 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_numDims()
136 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_dim()
Deager_operation_builder_jni.cc56 TFE_TensorHandle* requireTensorHandle(JNIEnv* env, jlong handle) { in requireTensorHandle()
62 return reinterpret_cast<TFE_TensorHandle*>(handle); in requireTensorHandle()
96 std::unique_ptr<TFE_TensorHandle*[]> retvals( in Java_org_tensorflow_EagerOperationBuilder_execute()
97 new TFE_TensorHandle*[num_retvals]); in Java_org_tensorflow_EagerOperationBuilder_execute()
132 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, input_handle); in Java_org_tensorflow_EagerOperationBuilder_addInput()
146 std::unique_ptr<TFE_TensorHandle*[]> tensor_handles( in Java_org_tensorflow_EagerOperationBuilder_addInputList()
147 new TFE_TensorHandle*[num_inputs]); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor_conversion.cc40 TFE_TensorHandle* TFE_TensorHandleCache::Lookup( in Lookup()
52 return new TFE_TensorHandle{ in Lookup()
58 TFE_TensorHandle* h) { in Insert()
62 new TFE_TensorHandle{ in Insert()
Dpywrap_tensor_conversion.h75 TFE_TensorHandle* Lookup(PyObject* value, tensorflow::DataType dtype,
79 absl::string_view device_name, TFE_TensorHandle* h);
96 absl::flat_hash_map<Key, TFE_TensorHandle*> cache;
Dpywrap_tfe.h30 typedef tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 4>
32 typedef tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2>
132 PyObject* EagerTensorFromHandle(TFE_TensorHandle* handle);
135 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o);
Dpywrap_tensor.cc77 PyObject* TFE_TensorHandleToNumpy(TFE_TensorHandle* handle, TF_Status* status) { in TFE_TensorHandleToNumpy()
199 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle, in EagerCast()
219 TFE_TensorHandle* output = nullptr; in EagerCast()
233 TFE_TensorHandle* ConvertToEagerTensorUncached(TFE_Context* ctx, in ConvertToEagerTensorUncached()
324 TFE_TensorHandle* ConvertToEagerTensor(TFE_Context* ctx, PyObject* value, in ConvertToEagerTensor()
333 TFE_TensorHandle* handle = cache->Lookup(value, dtype, device_name); in ConvertToEagerTensor()
359 TFE_TensorHandle* handle;
603 TFE_TensorHandle* handle = TFE_TensorHandleCopyToDevice( in EagerTensor_copy_to_device()
821 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o) { in EagerTensor_Handle()
825 PyObject* EagerTensorFromHandle(TFE_TensorHandle* handle) { in EagerTensorFromHandle()
[all …]
/external/tensorflow/tensorflow/python/lib/core/
Dsafe_ptr.h38 void operator()(TFE_TensorHandle* p) const { TFE_DeleteTensorHandle(p); } in operator()
64 std::unique_ptr<TFE_TensorHandle, detail::TFETensorHandleDeleter>;
65 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle);
Dpy_seq_tensor.cc279 TFE_TensorHandle** h, const char** error) { in Convert()
301 *h = new TFE_TensorHandle{std::make_unique<TensorHandleInterface>(handle)}; in Convert()
606 TFE_TensorHandle* NumpyToTFE_TensorHandle(TFE_Context* ctx, PyObject* obj) { in NumpyToTFE_TensorHandle()
622 return new TFE_TensorHandle{ in NumpyToTFE_TensorHandle()
631 TFE_TensorHandle* PySeqToTFE_TensorHandle(TFE_Context* ctx, PyObject* obj, in PySeqToTFE_TensorHandle()
703 TFE_TensorHandle* handle = nullptr; in PySeqToTFE_TensorHandle()
814 return new TFE_TensorHandle{std::make_unique<TensorHandleInterface>(h)}; in PySeqToTFE_TensorHandle()
Dpy_seq_tensor.h37 TFE_TensorHandle* PySeqToTFE_TensorHandle(TFE_Context* ctx, PyObject* obj,
Dsafe_ptr.cc28 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle) { in make_safe()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.cc129 TFE_TensorHandle* input_handle = TFE_NewTensorHandle(tf_tensor, status); in EvaluateOperation()
139 absl::InlinedVector<TFE_TensorHandle*, 2> outputs(num_outputs); in EvaluateOperation()
143 for (TFE_TensorHandle* tensor_handle : outputs) { in EvaluateOperation()
150 for (TFE_TensorHandle* tensor_handle : outputs) { in EvaluateOperation()
/external/tensorflow/tensorflow/go/
Dtensor_handle.go42 c *C.TFE_TensorHandle
63 func newTensorHandleFromC(c *C.TFE_TensorHandle) *TensorHandle { argument

12