Home
last modified time | relevance | path

Searched refs:TFE_ContextOptions (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api.h48 typedef struct TFE_ContextOptions TFE_ContextOptions; typedef
51 TF_CAPI_EXPORT extern TFE_ContextOptions* TFE_NewContextOptions(void);
58 TFE_ContextOptions* options, const void* proto, size_t proto_len,
78 TF_CAPI_EXPORT extern void TFE_ContextOptionsSetAsync(TFE_ContextOptions*,
82 TFE_ContextOptions*, TFE_ContextDevicePlacementPolicy);
85 TF_CAPI_EXPORT extern void TFE_DeleteContextOptions(TFE_ContextOptions*);
94 const TFE_ContextOptions* opts, TF_Status* status);
Dc_api_test.cc41 TFE_ContextOptions* opts = TFE_NewContextOptions(); in BM_InitOp()
64 TFE_ContextOptions* opts = TFE_NewContextOptions(); in BM_Execute()
93 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TEST()
148 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TestRemoteExecute()
232 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TestRemoteExecuteSilentCopies()
364 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TestRemoteExecuteChangeServerDef()
462 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TensorHandleCopyBetweenDevices()
539 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TensorHandleCopyBetweenDevicesError()
576 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TensorHandleCopyBetweenTwoGPUDevices()
655 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TensorHandleSilentCopy()
[all …]
Dc_api_experimental_test.cc41 TFE_ContextOptions* opts = TFE_NewContextOptions(); in ExecuteWithProfiling()
108 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TEST()
Dc_api.cc311 TFE_ContextOptions* TFE_NewContextOptions() { return new TFE_ContextOptions; } in TFE_NewContextOptions()
313 void TFE_ContextOptionsSetConfig(TFE_ContextOptions* options, const void* proto, in TFE_ContextOptionsSetConfig()
318 void TFE_ContextOptionsSetAsync(TFE_ContextOptions* options, in TFE_ContextOptionsSetAsync()
324 TFE_ContextOptions* options, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextOptionsSetDevicePlacementPolicy()
334 void TFE_DeleteContextOptions(TFE_ContextOptions* options) { delete options; } in TFE_DeleteContextOptions()
336 TFE_Context* TFE_NewContext(const TFE_ContextOptions* opts, TF_Status* status) { in TFE_NewContext()
353 TFE_Context* TFE_NewContextFromSession(const TFE_ContextOptions* opts, in TFE_NewContextFromSession()
Dc_api_internal.h58 struct TFE_ContextOptions { struct
/external/tensorflow/tensorflow/go/
Dcontext.go41 func (o *ContextOptions) c() (*C.TFE_ContextOptions, error) {
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc181 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TEST()
222 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TEST()
329 TFE_ContextOptions* opts = TFE_NewContextOptions(); in AddEagerOpToGraphTest()
Dc_api_experimental.h149 const TFE_ContextOptions* opts, TF_Session* sess, TF_Status* status);