Home
last modified time | relevance | path

Searched defs:eager_ctx (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dunified_api_testutil.h63 TFE_Context* eager_ctx = in TestTensorHandleWithDims() local
79 TFE_Context* eager_ctx = in TestScalarTensorHandle() local
Dc_api_unified_experimental_test.cc64 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
127 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
200 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
286 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
409 TFE_Context* eager_ctx = in TEST_P() local
501 TFE_Context* eager_ctx = in TEST_P() local
659 TFE_Context* eager_ctx = in TEST_P() local
848 TFE_Context* eager_ctx = in TEST_P() local
/external/tensorflow/tensorflow/core/function/
Druntime_client.h60 explicit Runtime(EagerContext& eager_ctx) : eager_ctx_(eager_ctx) {} in Runtime()
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
Dconversion_function.cc47 Device* op_device, EagerContext* eager_ctx) { in CopyRefGpuTensorToRuntimeFallbackTensor()
206 EagerContext* eager_ctx = eager_ctx_expected.get(); in ConvertDenseGpuTensorToRuntimeFallbackTensor() local
/external/tensorflow/tensorflow/core/tfrt/eager/
Dfunction_cache_test.cc170 TFE_Context* eager_ctx = in TestScalarTensorHandle() local
217 tensorflow::EagerContext* eager_ctx = tfrt_ctx->GetEagerContext(); in TEST_P() local
Dfunction_cache.cc48 tensorflow::EagerContext* eager_ctx, tfrt::CoreRuntime* corert, in GetOrAddFunction()
Dtransform_graph_function.cc38 EagerContext* eager_ctx, bool enable_grappler, in TransformGraphFunction()
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_kernels.cc367 EagerContext* eager_ctx) { in PrepareAttributes()
547 EagerContext* eager_ctx, const char* op_name, in CallEagerExecute()
584 const tfrt::ExecutionContext& exec_ctx, EagerContext* eager_ctx, in RuntimeFallbackExecute()
663 EagerContext* eager_ctx = eager_ctx_expected.get(); in RuntimeFallbackExecute() local
694 EagerContext* eager_ctx = eager_ctx_expected.get(); in RuntimeFallbackKernel() local
1114 EagerContext* eager_ctx = eager_ctx_expected.get(); in RuntimeFallbackExecuteOp() local
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dvariable.cc87 EagerContext* eager_ctx = reinterpret_cast<EagerContext*>(ctx); in CreateUninitialized() local