Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.h23 TFE_TensorHandle* TestScalarTensorHandle(float value);
26 TFE_TensorHandle* TestScalarTensorHandle(int value);
29 TFE_TensorHandle* TestScalarTensorHandle(bool value);
32 TFE_TensorHandle* DoubleTestMatrixTensorHandle();
35 TFE_TensorHandle* TestMatrixTensorHandle();
38 TFE_TensorHandle* DoubleTestMatrixTensorHandle3X2();
41 TFE_TensorHandle* TestMatrixTensorHandle3X2();
44 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
47 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a);
50 TFE_TensorHandle* TestAxisTensorHandle();
[all …]
Dc_api.h157 typedef struct TFE_TensorHandle TFE_TensorHandle; typedef
159 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t,
162 TF_CAPI_EXPORT extern void TFE_DeleteTensorHandle(TFE_TensorHandle* h);
163 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h);
165 TF_CAPI_EXPORT extern int TFE_TensorHandleNumDims(TFE_TensorHandle* h,
167 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h,
170 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleDim(TFE_TensorHandle* h,
180 TFE_TensorHandle* h, TF_Status* status);
186 TFE_TensorHandle* h, TF_Status* status);
191 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor(
[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_test.cc46 TFE_TensorHandle* m = TestMatrixTensorHandle(); in BM_InitOp()
70 TFE_TensorHandle* m = TestMatrixTensorHandle(); in BM_Execute()
72 TFE_TensorHandle* retvals[1]; in BM_Execute()
159 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(); in TestRemoteExecute()
160 TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(); in TestRemoteExecute()
174 TFE_TensorHandle* retvals[1]; in TestRemoteExecute()
242 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(); in TestRemoteExecuteSilentCopies()
243 TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(); in TestRemoteExecuteSilentCopies()
256 TFE_TensorHandle* retvals[1]; in TestRemoteExecuteSilentCopies()
300 void CheckTFE_TensorHandleHasFloats(TFE_TensorHandle* handle, in CheckTFE_TensorHandleHasFloats()
[all …]
Dc_api.cc237 TFE_TensorHandle* input) { in OpInferSingleInputAttrs()
260 TFE_TensorHandle** inputs, in OpInferSingleTypeInputListAttrs()
276 TFE_TensorHandle** inputs, int num_inputs) { in OpInferMixedTypeInputListAttrs()
292 tensorflow::Status OpInferInputListAttrs(TFE_Op* op, TFE_TensorHandle** inputs, in OpInferInputListAttrs()
423 TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t, TF_Status* status) { in TFE_NewTensorHandle()
427 return new TFE_TensorHandle(tensor, nullptr, nullptr); in TFE_NewTensorHandle()
430 void TFE_DeleteTensorHandle(TFE_TensorHandle* h) { in TFE_DeleteTensorHandle()
440 TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { in TFE_TensorHandleDataType()
444 int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleNumDims()
455 int64_t TFE_TensorHandleNumElements(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleNumElements()
[all …]
Dc_api_internal.h78 struct TFE_TensorHandle { struct
79 TFE_TensorHandle(const tensorflow::Tensor& t, tensorflow::Device* d, in TFE_TensorHandle() function
83 TFE_TensorHandle(tensorflow::TensorHandle* handle) : handle(handle) {} in TFE_TensorHandle() argument
88 TFE_TensorHandle(TF_Output t, TF_DataType dtype) in TFE_TensorHandle() argument
Dc_api_debug_test.cc24 TFE_TensorHandle* h = TestScalarTensorHandle(1.0f); in TEST()
37 TFE_TensorHandle* h = TestMatrixTensorHandle3X2();
Dc_api_debug.cc31 std::vector<int64> TensorShapeAsVector(TFE_TensorHandle* handle, in TensorShapeAsVector()
53 TFE_TensorHandle* handle, TF_Status* status) { in TFE_TensorHandleTensorDebugInfo()
Dc_api_experimental_test.cc51 TFE_TensorHandle* m = TestMatrixTensorHandle(); in ExecuteWithProfiling()
53 TFE_TensorHandle* retvals[1] = {nullptr}; in ExecuteWithProfiling()
Dc_api_experimental.h25 TF_CAPI_EXPORT extern void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h,
Dc_api_experimental.cc25 void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h, TF_Status* status) { in TFE_OpConsumeInput()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.h164 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_DequeueNamedTensor(
168 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(
173 TFE_TensorHandle* tensor,
177 TFE_Context* ctx, int tensor_id, TFE_TensorHandle* tensor,
183 TFE_TensorHandle* tensor,
186 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_DequeueVariantTensor(
191 TFE_TensorHandle* handle);
206 TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals,
258 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromScalar(
271 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromTFOutput(
[all …]
Dc_api_experimental_test.cc186 TFE_TensorHandle* m = TestMatrixTensorHandle(); in TEST()
190 TFE_TensorHandle* retvals[1] = {nullptr}; in TEST()
229 TFE_TensorHandle* m = TestMatrixTensorHandle(); in TEST()
264 TFE_TensorHandle* send_retvals; in TEST()
269 TFE_TensorHandle* recv_retvals[1] = {nullptr}; in TEST()
313 TFE_TensorHandle* m = TestMatrixTensorHandle(); in TEST()
345 TFE_TensorHandle* retvals[5]; in AddEagerOpToGraphAndCheck()
366 TFE_TensorHandle* m = TestMatrixTensorHandle(); in TEST_F()
372 TFE_TensorHandle* retvals[5]; in TEST_F()
399 TFE_TensorHandle* axis = TestAxisTensorHandle(); in TEST_F()
[all …]
Dc_api_experimental.cc8570 static TFE_TensorHandle* createTFEQueue(TFE_Context* ctx, TF_DataType inputType, in createTFEQueue()
8592 TFE_TensorHandle* queue = nullptr; in createTFEQueue()
8601 TFE_TensorHandle* queue, TFE_TensorHandle* tensor, in createTFEEnqueue()
8621 static TFE_TensorHandle* createTFEDequeue(TFE_Context* ctx, in createTFEDequeue()
8623 TFE_TensorHandle* queue, in createTFEDequeue()
8635 TFE_TensorHandle* ret; in createTFEDequeue()
8643 TFE_TensorHandle* TFE_DequeueNamedTensor(TF_Session* session, int tensor_id, in TFE_DequeueNamedTensor()
8654 TFE_TensorHandle* queue = createTFEQueue(ctx, inputType, tensor_id, status); in TFE_DequeueNamedTensor()
8656 std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> in TFE_DequeueNamedTensor()
8663 TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_DequeueNamedTensorFromCtx()
[all …]
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.h30 TFE_TensorHandle* ConvertToEagerTensor(PyObject* value, PyObject* dtype);
34 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle,
Dpywrap_tfe.h30 typedef tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 4>
32 typedef tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2>
118 PyObject* EagerTensorFromHandle(TFE_TensorHandle* handle);
121 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o);
Dpywrap_tensor.cc59 TFE_TensorHandle* NumpyToTensorHandle(PyObject* obj) { in NumpyToTensorHandle()
74 TFE_TensorHandle* CopyToDevice(TFE_TensorHandle* handle, PyObject* ctx, in CopyToDevice()
96 TFE_TensorHandle* new_handle = in CopyToDevice()
176 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle, in EagerCast()
196 TFE_TensorHandle* output = nullptr; in EagerCast()
211 TFE_TensorHandle* ConvertToEagerTensor(PyObject* value, PyObject* dtype) { in ConvertToEagerTensor()
287 TFE_TensorHandle* handle;
418 tensorflow::make_safe(static_cast<TFE_TensorHandle*>( in EagerTensor_init()
819 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o) { in EagerTensor_Handle()
823 PyObject* EagerTensorFromHandle(TFE_TensorHandle* handle) { in EagerTensorFromHandle()
[all …]
Dpywrap_tfe_src.cc1467 TFE_TensorHandle* t = EagerTensor_Handle(tensor);
2238 TFE_TensorHandle* output_handle;
2302 tensorflow::make_safe(static_cast<TFE_TensorHandle*>(
2379 TFE_TensorHandle* input_handle = EagerTensor_Handle(py_eager_tensor.get());
2715 TFE_TensorHandle* input_handle =
2766 tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2> retvals(num_retvals);
2896 TFE_TensorHandle* t = EagerTensor_Handle(arg);
/external/tensorflow/tensorflow/python/lib/core/
Dsafe_ptr.h38 void operator()(TFE_TensorHandle* p) const { TFE_DeleteTensorHandle(p); } in operator()
60 std::unique_ptr<TFE_TensorHandle, detail::TFETensorHandleDeleter>;
61 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle);
Dsafe_ptr.cc28 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle) { in make_safe()
Dpy_func.cc84 arg = EagerTensorFromHandle(new TFE_TensorHandle(t, device, device)); in MakeArgTuple()
/external/tensorflow/tensorflow/go/
Dtensor_handle.go42 c *C.TFE_TensorHandle
63 func newTensorHandleFromC(c *C.TFE_TensorHandle) *TensorHandle { argument