Home
last modified time | relevance | path

Searched refs:TFE_Context (Results 1 – 25 of 67) sorted by relevance

123

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.h24 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, float value);
27 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, int value);
30 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, bool value);
33 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx,
37 TFE_TensorHandle* DoubleTestMatrixTensorHandle(TFE_Context* ctx);
40 TFE_TensorHandle* TestMatrixTensorHandle(TFE_Context* ctx);
44 TFE_TensorHandle* TestMatrixTensorHandleWithInput(TFE_Context* ctx,
49 TFE_TensorHandle* TestTensorHandleWithDimsFloat(TFE_Context* ctx, float data[],
53 TFE_TensorHandle* TestTensorHandleWithDimsInt(TFE_Context* ctx, int data[],
57 TFE_TensorHandle* TestMatrixTensorHandle100x100(TFE_Context* ctx);
[all …]
Dc_api.h97 typedef struct TFE_Context TFE_Context; typedef
99 TF_CAPI_EXPORT extern TFE_Context* TFE_NewContext(
101 TF_CAPI_EXPORT extern void TFE_DeleteContext(TFE_Context* ctx);
102 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
107 TF_CAPI_EXPORT extern void TFE_ContextClearCaches(TFE_Context* ctx);
114 TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy);
119 TFE_ContextGetDevicePlacementPolicy(TFE_Context* ctx);
127 TF_CAPI_EXPORT extern void TFE_ContextSetServerDef(TFE_Context* ctx,
193 TFE_TensorHandle* h, TFE_Context* ctx, const char* device_name,
248 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
[all …]
Dc_api_experimental.h39 TF_CAPI_EXPORT extern void TFE_ContextEnableGraphCollection(TFE_Context* ctx);
43 TF_CAPI_EXPORT extern void TFE_ContextDisableGraphCollection(TFE_Context* ctx);
278 TF_CAPI_EXPORT extern uint64_t TFE_GetContextId(TFE_Context* ctx);
338 TF_CAPI_EXPORT extern void TFE_ContextSetExecutorForThread(TFE_Context*,
343 TFE_Context*);
354 TF_CAPI_EXPORT extern void TFE_ContextUpdateServerDef(TFE_Context* ctx,
362 TF_CAPI_EXPORT extern bool TFE_ContextCheckAlive(TFE_Context* ctx,
369 TF_CAPI_EXPORT extern void TFE_ContextAsyncWait(TFE_Context* ctx,
391 TFE_Context* ctx, const char* device_name, TF_DataType, const int64_t* dims,
398 TF_CAPI_EXPORT extern void TFE_HostAddressSpace(TFE_Context* ctx,
[all …]
Dc_api_test_util.cc31 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, float value) { in TestScalarTensorHandle()
43 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, in TestScalarTensorHandle()
56 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, int value) { in TestScalarTensorHandle()
68 TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, bool value) { in TestScalarTensorHandle()
80 TFE_TensorHandle* DoubleTestMatrixTensorHandle(TFE_Context* ctx) { in DoubleTestMatrixTensorHandle()
94 TFE_TensorHandle* TestMatrixTensorHandle(TFE_Context* ctx) { in TestMatrixTensorHandle()
108 TFE_TensorHandle* TestMatrixTensorHandleWithInput(TFE_Context* ctx, in TestMatrixTensorHandleWithInput()
122 TFE_TensorHandle* TestTensorHandleWithDimsFloat(TFE_Context* ctx, float data[], in TestTensorHandleWithDimsFloat()
135 TFE_TensorHandle* TestTensorHandleWithDimsInt(TFE_Context* ctx, int data[], in TestTensorHandleWithDimsInt()
148 TFE_TensorHandle* TestMatrixTensorHandle100x100(TFE_Context* ctx) { in TestMatrixTensorHandle100x100()
[all …]
Dtfe_context_internal.h27 typedef struct TFE_Context TFE_Context; typedef
31 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionContext, TFE_Context);
Dc_api.cc115 TFE_Context* TFE_NewContext(const TFE_ContextOptions* opts, TF_Status* status) { in TFE_NewContext()
158 void TFE_DeleteContext(TFE_Context* ctx) { in TFE_DeleteContext()
167 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()
173 void TFE_ContextClearCaches(TFE_Context* ctx) { in TFE_ContextClearCaches()
178 TF_CAPI_EXPORT extern void TFE_ContextSetServerDef(TFE_Context* ctx, in TFE_ContextSetServerDef()
199 TF_CAPI_EXPORT extern void TFE_ContextUpdateServerDef(TFE_Context* ctx, in TFE_ContextUpdateServerDef()
226 TF_CAPI_EXPORT extern bool TFE_ContextCheckAlive(TFE_Context* ctx, in TFE_ContextCheckAlive()
242 TF_CAPI_EXPORT extern void TFE_ContextAsyncWait(TFE_Context* ctx, in TFE_ContextAsyncWait()
252 TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextSetThreadLocalDevicePlacementPolicy()
261 TFE_Context* ctx) { in TFE_ContextGetDevicePlacementPolicy()
[all …]
Dtfe_op_attrs_internal.h27 typedef struct TFE_Context TFE_Context; typedef
34 void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op,
Dc_api_experimental.cc51 void TFE_ContextEnableGraphCollection(TFE_Context* ctx) { in TFE_ContextEnableGraphCollection()
55 void TFE_ContextDisableGraphCollection(TFE_Context* ctx) { in TFE_ContextDisableGraphCollection()
59 uint64_t TFE_GetContextId(TFE_Context* ctx) { in TFE_GetContextId()
536 void TFE_ContextSetExecutorForThread(TFE_Context* ctx, TFE_Executor* executor) { in TFE_ContextSetExecutorForThread()
540 TFE_Executor* TFE_ContextGetExecutorForThread(TFE_Context* ctx) { in TFE_ContextGetExecutorForThread()
544 void TFE_HostAddressSpace(TFE_Context* ctx, TF_Buffer* buf) { in TFE_HostAddressSpace()
557 void TFE_ContextGetFunctionDef(TFE_Context* ctx, const char* function_name, in TFE_ContextGetFunctionDef()
576 TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype, in TFE_AllocateHostTensor()
602 TFE_TensorHandle* TFE_NewTensorHandleFromTensor(TFE_Context* ctx, TF_Tensor* t, in TFE_NewTensorHandleFromTensor()
608 TFE_TensorHandle* TFE_CreatePackedTensorHandle(TFE_Context* ctx, in TFE_CreatePackedTensorHandle()
[all …]
Dcustom_device_testutil.cc53 TFE_Context* context, const tensorflow::string& logging_device_name, in MakeLoggedTensorHandle()
67 TFE_TensorHandle* CopyToLoggingDevice(TFE_Context* context, in CopyToLoggingDevice()
80 TFE_TensorHandle* CopyTensorFromLoggingDevice(TFE_Context* context, in CopyTensorFromLoggingDevice()
103 TFE_Context* context = TFE_OpGetContext(original_op, s); in LoggingDeviceExecute()
154 void RegisterLoggingDevice(TFE_Context* context, const char* name, in RegisterLoggingDevice()
Dc_api_test.cc56 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_InitOp()
80 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_Execute()
122 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_Execute_Identity()
157 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
180 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TEST()
205 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevices()
282 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevicesError()
322 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
405 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleSilentCopy()
481 TFE_Context* ctx = TFE_NewContext(opts, status); in SetAndGetOpDevices()
[all …]
Dgradients_util.cc40 Status ScalarTensorHandleHelper(TFE_Context* ctx, float value, in ScalarTensorHandleHelper()
54 Status TensorHandleWithDimsFloatHelper(TFE_Context* ctx, float data[], in TensorHandleWithDimsFloatHelper()
68 Status TensorHandleWithDimsIntHelper(TFE_Context* ctx, int data[], in TensorHandleWithDimsIntHelper()
87 TFE_Context* eager_ctx = in ScalarTensorHandle()
103 TFE_Context* eager_ctx = in TensorHandleWithDimsFloat()
119 TFE_Context* eager_ctx = in TensorHandleWithDimsInt()
Dc_api_debug_test.cc26 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
45 TFE_Context* ctx = TFE_NewContext(opts, status);
Dc_api_unified_experimental_eager.cc48 TFE_Context* c_ctx = TFE_NewContext(options, s); in TF_NewEagerExecutionContext()
72 TFE_Context* TF_ExecutionContextGetTFEContext(TF_ExecutionContext* ctx, in TF_ExecutionContextGetTFEContext()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.h46 static Variable* Create(TFE_Context* context, TF_DataType type,
52 void Destroy(TFE_Context* context, TF_Status* status);
55 TensorHandlePtr Read(TFE_Context* context, TF_Status* status);
57 void Assign(TFE_Context* context, TFE_TensorHandle* value, TF_Status* status);
59 void AssignAdd(TFE_Context* context, TFE_TensorHandle* value,
65 void GeneralAssignment(const char* op_name, TFE_Context* context,
86 TFE_Context* context, TFE_TensorHandle* input,
92 TFE_Context* context,
96 TensorHandlePtr Multiply(TFE_Context* context, TFE_TensorHandle* first,
105 TFE_Context* context, const char* device_name,
[all …]
Dparallel_device_lib.h67 std::unique_ptr<ParallelTensor> CopyToParallelDevice(TFE_Context* context,
74 absl::Span<const DataType> values, TFE_Context* context,
78 std::unique_ptr<ParallelTensor> DeviceIDs(TFE_Context* context,
96 TFE_Context* context, const std::vector<ParallelTensor*>& inputs,
112 void StartExecute(TFE_Context* context,
211 absl::Span<DataType const> values, TFE_Context* context, in ScalarsFromSequence()
Dparallel_device_testlib.cc34 Variable* Variable::Create(TFE_Context* context, TF_DataType type, in Create()
60 void Variable::Destroy(TFE_Context* context, TF_Status* status) { in Destroy()
78 TensorHandlePtr Variable::Read(TFE_Context* context, TF_Status* status) { in Read()
96 void Variable::GeneralAssignment(const char* op_name, TFE_Context* context, in GeneralAssignment()
115 void Variable::AssignAdd(TFE_Context* context, TFE_TensorHandle* value, in AssignAdd()
120 void Variable::Assign(TFE_Context* context, TFE_TensorHandle* value, in Assign()
160 TFE_Context* context, TFE_TensorHandle* input, in ExtractPerDeviceValues()
182 TensorHandlePtr Multiply(TFE_Context* context, TFE_TensorHandle* first, in Multiply()
204 void BasicTestsForTwoDevices(TFE_Context* context, const char* first_device, in BasicTestsForTwoDevices()
Dparallel_device_lib.cc88 void StartExecute(TFE_Context* context, const char* operation_name,
99 void Execute(TFE_Context* context, const char* operation_name,
126 TFE_Context* context_ TF_GUARDED_BY(execution_mutex_);
175 void DeviceThread::StartExecute(TFE_Context* context, in StartExecute()
221 void DeviceThread::Execute(TFE_Context* context, const char* operation_name, in Execute()
273 TFE_Context* context, TFE_TensorHandle* tensor, TF_Status* status) const { in CopyToParallelDevice()
287 TFE_Context* context, TF_Status* status) const { in DeviceIDs()
297 ParallelDevice::Execute(TFE_Context* context, in Execute()
321 TFE_Context* context, const std::vector<ParallelTensor*>& inputs, in StartExecute()
/external/tensorflow/tensorflow/cc/experimental/base/public/
Druntime.h47 explicit Runtime(TFE_Context* ctx) : ctx_(ctx) {} in Runtime()
51 void Reset(TFE_Context* ctx) { ctx_.reset(ctx); } in Reset()
55 TFE_Context* GetTFEContext() const { return ctx_.get(); } in GetTFEContext()
62 void operator()(TFE_Context* p) const { TFE_DeleteContext(p); } in operator()
64 std::unique_ptr<TFE_Context, TFEContextDeleter> ctx_;
/external/tensorflow/tensorflow/python/lib/core/
Dpy_seq_tensor.cc281 static Status Convert(TFE_Context* ctx, PyObject* obj, ConverterState* state, in Convert()
319 static AbstractTensorInterface* CreateScalar(TFE_Context* ctx, int64 value) { in CreateScalar()
324 TFE_Context* ctx, absl::Span<const int64> dim_sizes) { in CreateTensor()
359 static AbstractTensorInterface* CreateScalar(TFE_Context* ctx, uint64 value) { in CreateScalar()
364 TFE_Context* ctx, absl::Span<const int64> dim_sizes) { in CreateTensor()
396 static AbstractTensorInterface* CreateScalar(TFE_Context* ctx, int32 value) { in CreateScalar()
401 TFE_Context* ctx, absl::Span<const int64> dim_sizes) { in CreateTensor()
503 static AbstractTensorInterface* CreateScalar(TFE_Context* ctx, float value) { in CreateScalar()
508 TFE_Context* ctx, absl::Span<const int64> dim_sizes) { in CreateTensor()
519 static AbstractTensorInterface* CreateScalar(TFE_Context* ctx, double value) { in CreateScalar()
[all …]
Dpy_func.cc88 Status MakeArgTuple(const PyCall* call, TFE_Context* ctx, PyObject** tuple) { in MakeArgTuple()
151 TFE_Context* ctx, in ExtractTensorFromEagerTensor()
200 TFE_Context* ctx = reinterpret_cast<TFE_Context*>(PyCapsule_GetPointer( in DoCallPyFunc()
239 TFE_Context* ctx = reinterpret_cast<TFE_Context*>(PyCapsule_GetPointer( in DoCallPyFunc()
/external/tensorflow/tensorflow/java/src/main/native/
Deager_session_jni.cc44 TFE_Context* context = TFE_NewContext(opts, status); in Java_org_tensorflow_EagerSession_allocate()
54 static_assert(sizeof(jlong) >= sizeof(TFE_Context*), in Java_org_tensorflow_EagerSession_allocate()
63 TFE_DeleteContext(reinterpret_cast<TFE_Context*>(handle)); in Java_org_tensorflow_EagerSession_delete()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor_conversion.h76 TFE_Context* ctx,
79 void Insert(PyObject* value, tensorflow::DataType dtype, TFE_Context* ctx,
88 using Key = std::tuple<PyObjectPtr, tensorflow::DataType, TFE_Context*,
Dpywrap_tensor_conversion.cc41 PyObject* value, tensorflow::DataType dtype, TFE_Context* ctx, in Lookup()
56 TFE_Context* ctx, in Insert()
/external/tensorflow/tensorflow/python/framework/
Dpython_tensor_converter.h45 PythonTensorConverter(PyObject* py_eager_context, TFE_Context* ctx, in PythonTensorConverter()
70 TFE_Context* ctx_;
/external/tensorflow/tensorflow/c/experimental/saved_model/public/
Dsaved_model_api.h46 TFE_Context* ctx,
63 const char* dirname, TFE_Context* ctx, const char* const* tags,

123