| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | partial_run_mgr_test.cc | 28 CancellationManager* cancellation_manager; in TEST() local
37 CancellationManager* cancellation_manager; in TEST() local
49 CancellationManager* cancellation_manager; in TEST() local
63 CancellationManager* cancellation_manager; in TEST() local
111 CancellationManager* cancellation_manager; in TEST_P() local
127 CancellationManager* cancellation_manager; in TEST_P() local
|
| D | partial_run_mgr.h | 72 std::unique_ptr<CancellationManager> cancellation_manager; member
|
| D | partial_run_mgr.cc | 23 CancellationManager** cancellation_manager) { in FindOrCreate()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 259 CancellationManager cancellation_manager; member
266 CancellationManager* cancellation_manager, in Run()
354 CancellationManager* cancellation_manager, in PrepareForRun()
415 CancellationManager* cancellation_manager, in Run()
460 CancellationManager* cancellation_manager, in RunAsync()
|
| D | execute_node.h | 99 CancellationManager* cancellation_manager, in ExecuteNode()
155 CancellationManager* cancellation_manager, in AsyncExecuteNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.cc | 90 CancellationManager* cancellation_manager) { in ProvideBuf()
152 CancellationManager* cancellation_manager) { in ConsumeBuf()
|
| D | collective_rma_local.cc | 31 CancellationManager* cancellation_manager, const StatusCallback& done) { in RecvFromPeer()
102 CancellationManager* cancellation_manager, const StatusCallback& done) { in PostToPeer()
|
| D | collective_test_util.cc | 63 CancellationManager* cancellation_manager, const StatusCallback& done) { in RecvFromPeer()
77 CancellationManager* cancellation_manager, in PostToPeer()
318 CancellationManager cancellation_manager; in RunCollective() local
|
| D | buf_rendezvous.h | 70 CancellationManager* cancellation_manager; member
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data.h | 62 tensorflow::CancellationManager* cancellation_manager) { in SetCancellationManager()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 323 OpKernelContext* ctx, CancellationManager* cancellation_manager, in RegisterCancellation()
363 OpKernelContext* ctx, CancellationManager* cancellation_manager, in UnregisterCancellation()
423 CancellationManager* cancellation_manager, OpKernelContext* ctx, in TPUExecute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | reduce_dataset_op.cc | 69 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
|
| D | iterator_ops.h | 118 CancellationManager* cancellation_manager() { in cancellation_manager() function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model_test.cc | 1307 CancellationManager cancellation_manager; in TEST_P() local
2164 CancellationManager cancellation_manager; in TEST_F() local
2216 CancellationManager cancellation_manager; in TEST_F() local
2283 CancellationManager cancellation_manager; in TEST_F() local
2352 CancellationManager cancellation_manager; in TEST_F() local
2421 CancellationManager cancellation_manager; in TEST_F() local
|
| D | cancellation.cc | 236 Status RegisterCancellationCallback(CancellationManager* cancellation_manager, in RegisterCancellationCallback()
|
| D | rendezvous.h | 54 CancellationManager* cancellation_manager = nullptr; // not owned. member
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_compat_request_state.h | 125 CancellationManager* cancellation_manager() const { in cancellation_manager() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | to_tf_record_op.cc | 84 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator_base.cc | 111 CancellationManager* cancellation_manager, CancellationToken token) { in Cancel()
|
| D | conditional_accumulator_base.h | 96 CancellationManager* cancellation_manager; // not owned member
|
| D | queue_base.h | 159 CancellationManager* cancellation_manager; // not owned member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 44 CancellationManager* cancellation_manager, in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 441 def call(self, ctx, args, cancellation_manager=None): argument
1576 def _call_impl(self, args, kwargs, cancellation_manager=None): argument
1594 def _call_with_flat_signature(self, args, kwargs, cancellation_manager): argument
1643 def _call_with_structured_signature(self, args, kwargs, cancellation_manager): argument
1756 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
1851 def _experimental_with_cancellation_manager(self, cancellation_manager): argument
|
| /external/tensorflow/tensorflow/core/data/ |
| D | captured_function.cc | 801 CancellationManager cancellation_manager(ctx->cancellation_manager()); in Run() local
864 CancellationManager cancellation_manager(ctx->cancellation_manager()); in RunWithBorrowedArgs() local
919 CancellationManager cancellation_manager; in RunInstantiated() local
966 auto cancellation_manager = in RunAsync() local
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 197 CancellationManager& cancellation_manager, in StartExecute()
351 CancellationManager& cancellation_manager, in StartExecute()
|