Searched refs:TFE_NewContext (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 43 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_InitOp()
67 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_Execute()
102 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_Execute_Identity()
133 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
175 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevices()
252 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevicesError()
292 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
372 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleSilentCopy()
418 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleSilentCopyLocal()
463 TFE_Context* ctx = TFE_NewContext(opts, status); in SetAndGetOpDevices()
[all …]
|
| D | c_api_remote_test.cc | 69 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecute()
154 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteSilentCopies()
240 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
361 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteChangeServerDef()
|
| D | c_api_experimental_test.cc | 46 TFE_Context* ctx = TFE_NewContext(opts, status); in ExecuteWithProfiling()
321 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
397 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
456 TFE_Context* ctx = TFE_NewContext(opts, status); in Executor_MatMul_CPU()
505 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
|
| D | c_api.h | 99 TF_CAPI_EXPORT extern TFE_Context* TFE_NewContext(
|
| D | c_api.cc | 698 TFE_Context* TFE_NewContext(const TFE_ContextOptions* opts, TF_Status* status) { in TFE_NewContext() function
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 90 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
132 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
212 tfe_context_ = TFE_NewContext(tfe_context_options_, status_); in ShapeInferenceTest()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | constant_fold.cc | 44 auto ctx = TFE_NewContext(opts, status); in ConstantFoldFallbackHook()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | eager_session_jni.cc | 44 TFE_Context* context = TFE_NewContext(opts, status); in Java_org_tensorflow_EagerSession_allocate()
|
| /external/tensorflow/tensorflow/go/ |
| D | context.go | 86 cContext := C.TFE_NewContext(cOpt, status.c)
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 349 TFE_Context* context = TFE_NewContext(opts, status.get()); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 513 context_handle = pywrap_tfe.TFE_NewContext(opts)
|