Searched refs:enable_streaming_enqueue (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | tfe_executor_internal.h | 23 explicit TFE_Executor(bool async, bool enable_streaming_enqueue) in TFE_Executor()
25 new tensorflow::EagerExecutor(async, enable_streaming_enqueue)) {} in TFE_Executor()
|
| H A D | c_api_experimental.cc | 573 TFE_Executor* TFE_NewExecutor(bool is_async, bool enable_streaming_enqueue) { in TFE_NewExecutor() argument
574 return new TFE_Executor(is_async, enable_streaming_enqueue); in TFE_NewExecutor()
|
| H A D | c_api_experimental.h | 337 bool is_async, bool enable_streaming_enqueue);
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | executor.py | 72 def new_executor(enable_async, enable_streaming_enqueue=True): argument
73 handle = pywrap_tfe.TFE_NewExecutor(enable_async, enable_streaming_enqueue)
|
| H A D | remote_test.py | 147 enable_async=False, enable_streaming_enqueue=False)):
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| H A D | grpc_eager_client.cc | 191 void StreamingEnqueueAsync(bool enable_streaming_enqueue, in StreamingEnqueueAsync() argument
201 if (EnableStreaming() && enable_streaming_enqueue) { in StreamingEnqueueAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | eager_executor.cc | 34 EagerExecutor::EagerExecutor(bool async, bool enable_streaming_enqueue) in EagerExecutor() argument
44 enable_streaming_enqueue_(enable_streaming_enqueue) {} in EagerExecutor()
|
| H A D | eager_executor.h | 114 explicit EagerExecutor(bool async, bool enable_streaming_enqueue = true);
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | eager_client.h | 66 virtual void StreamingEnqueueAsync(bool enable_streaming_enqueue,
|
| H A D | eager_service_impl_test.cc | 107 void StreamingEnqueueAsync(bool enable_streaming_enqueue, in StreamingEnqueueAsync() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ |
| H A D | tfe_wrapper.cc | 941 [](const bool is_async, const bool enable_streaming_enqueue) { in PYBIND11_MODULE() argument
942 TFE_Executor* exc = TFE_NewExecutor(is_async, enable_streaming_enqueue); in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | checkpoint.py | 1267 enable_async=False, enable_streaming_enqueue=False)):
|