Searched refs:TFE_ContextOptions (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/c/eager/ |
D | c_api.h | 48 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);
|
D | c_api_test.cc | 41 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 …]
|
D | c_api_experimental_test.cc | 41 TFE_ContextOptions* opts = TFE_NewContextOptions(); in ExecuteWithProfiling() 108 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TEST()
|
D | c_api.cc | 311 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()
|
D | c_api_internal.h | 58 struct TFE_ContextOptions { struct
|
/external/tensorflow/tensorflow/go/ |
D | context.go | 41 func (o *ContextOptions) c() (*C.TFE_ContextOptions, error) {
|
/external/tensorflow/tensorflow/c/ |
D | c_api_experimental_test.cc | 181 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TEST() 222 TFE_ContextOptions* opts = TFE_NewContextOptions(); in TEST() 329 TFE_ContextOptions* opts = TFE_NewContextOptions(); in AddEagerOpToGraphTest()
|
D | c_api_experimental.h | 149 const TFE_ContextOptions* opts, TF_Session* sess, TF_Status* status);
|