/external/tensorflow/tensorflow/c/eager/ |
D | c_api_test_util.h | 23 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 …]
|
D | c_api.h | 157 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 …]
|
D | c_api_test_util.cc | 24 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 …]
|
D | c_api_test.cc | 46 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 …]
|
D | c_api.cc | 237 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 …]
|
D | c_api_internal.h | 78 struct TFE_TensorHandle { struct 79 TFE_TensorHandle(const tensorflow::Tensor& t, tensorflow::Device* d, in TFE_TensorHandle() argument 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
|
D | c_api_debug_test.cc | 24 TFE_TensorHandle* h = TestScalarTensorHandle(1.0f); in TEST() 37 TFE_TensorHandle* h = TestMatrixTensorHandle3X2();
|
D | c_api_debug.cc | 31 std::vector<int64> TensorShapeAsVector(TFE_TensorHandle* handle, in TensorShapeAsVector() 53 TFE_TensorHandle* handle, TF_Status* status) { in TFE_TensorHandleTensorDebugInfo()
|
D | c_api_experimental_test.cc | 51 TFE_TensorHandle* m = TestMatrixTensorHandle(); in ExecuteWithProfiling() 53 TFE_TensorHandle* retvals[1] = {nullptr}; in ExecuteWithProfiling()
|
D | c_api_experimental.h | 25 TF_CAPI_EXPORT extern void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h,
|
D | c_api_experimental.cc | 25 void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h, TF_Status* status) { in TFE_OpConsumeInput()
|
/external/tensorflow/tensorflow/c/ |
D | c_api_experimental.h | 164 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 …]
|
D | c_api_experimental_test.cc | 186 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 …]
|
D | c_api_experimental.cc | 8570 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/ |
D | pywrap_tensor.h | 30 TFE_TensorHandle* ConvertToEagerTensor(PyObject* value, PyObject* dtype); 34 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle,
|
D | pywrap_tfe.h | 30 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);
|
D | pywrap_tensor.cc | 59 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 …]
|
D | pywrap_tfe_src.cc | 1467 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/ |
D | safe_ptr.h | 38 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);
|
D | safe_ptr.cc | 28 Safe_TFE_TensorHandlePtr make_safe(TFE_TensorHandle* handle) { in make_safe()
|
D | py_func.cc | 84 arg = EagerTensorFromHandle(new TFE_TensorHandle(t, device, device)); in MakeArgTuple()
|
/external/tensorflow/tensorflow/go/ |
D | tensor_handle.go | 42 c *C.TFE_TensorHandle 63 func newTensorHandleFromC(c *C.TFE_TensorHandle) *TensorHandle { argument
|