| /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 | 238 CancellationManager cancellation_manager; member
245 CancellationManager* cancellation_manager, in Run()
324 CancellationManager* cancellation_manager, in Run()
341 CancellationManager* cancellation_manager, in RunAsync()
|
| D | execute_node.h | 99 CancellationManager* cancellation_manager, in ExecuteNode()
155 CancellationManager* cancellation_manager, in AsyncExecuteNode()
|
| D | eager_operation.h | 180 void SetCancellationManager(CancellationManager* cancellation_manager) { in SetCancellationManager()
|
| /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 | buf_rendezvous.h | 70 CancellationManager* cancellation_manager; member
|
| D | executor.h | 100 CancellationManager* cancellation_manager = nullptr; member
|
| D | graph_runner.cc | 185 CancellationManager cancellation_manager; in Run() local
|
| D | permuter_test.cc | 80 CancellationManager* cancellation_manager, in RecvFromPeer()
95 CancellationManager* cancellation_manager, in PostToPeer()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 321 OpKernelContext* ctx, CancellationManager* cancellation_manager, in RegisterCancellation()
359 OpKernelContext* ctx, CancellationManager* cancellation_manager, in UnregisterCancellation()
419 CancellationManager* cancellation_manager, OpKernelContext* ctx, in TPUExecute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | to_tf_record_op.cc | 82 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 | 158 CancellationManager* cancellation_manager; // not owned member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 43 CancellationManager* cancellation_manager, in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | cancellation.cc | 209 Status RegisterCancellationCallback(CancellationManager* cancellation_manager, in RegisterCancellationCallback()
|
| D | rendezvous.h | 54 CancellationManager* cancellation_manager = nullptr; // not owned. member
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 108 CancellationManager* cancellation_manager() { in cancellation_manager() function
|
| D | captured_function.cc | 849 CancellationManager cancellation_manager(ctx->cancellation_manager()); in Run() local
912 CancellationManager cancellation_manager(ctx->cancellation_manager()); in RunWithBorrowedArgs() local
967 CancellationManager cancellation_manager; in RunInstantiated() local
1013 auto cancellation_manager = in RunAsync() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental.cc | 494 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerStartCancel()
499 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerIsCancelled()
504 TFE_CancellationManager* cancellation_manager) { in TFE_DeleteCancellationManager()
509 TFE_CancellationManager* cancellation_manager, in TFE_OpSetCancellationManager()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | execute.py | 85 cancellation_manager, argument
|
| D | function.py | 537 def call(self, ctx, args, cancellation_manager=None): argument
1695 def _call_impl(self, args, kwargs, cancellation_manager=None): argument
1713 def _call_with_flat_signature(self, args, kwargs, cancellation_manager): argument
1762 def _call_with_structured_signature(self, args, kwargs, cancellation_manager): argument
1868 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
1960 def _experimental_with_cancellation_manager(self, cancellation_manager): argument
|