| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | partial_run_mgr_test.cc | 28 CancellationManager* cancellation_manager; in TEST() local
29 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager); in TEST()
30 EXPECT_TRUE(cancellation_manager != nullptr); in TEST()
37 CancellationManager* cancellation_manager; in TEST() local
38 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager); in TEST()
42 EXPECT_EQ(cancellation_manager, found_cancellation_manager); in TEST()
49 CancellationManager* cancellation_manager; in TEST() local
50 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager); in TEST()
55 EXPECT_NE(cancellation_manager, new_cancellation_manager); in TEST()
63 CancellationManager* cancellation_manager; in TEST() local
[all …]
|
| D | partial_run_mgr.cc | 23 CancellationManager** cancellation_manager) { in FindOrCreate() argument
27 *cancellation_manager = it->second->cancellation_manager.get(); in FindOrCreate()
33 partial_run->cancellation_manager = in FindOrCreate()
35 *cancellation_manager = partial_run->cancellation_manager.get(); in FindOrCreate()
|
| D | partial_run_mgr.h | 53 bool FindOrCreate(int step_id, CancellationManager** cancellation_manager);
72 std::unique_ptr<CancellationManager> cancellation_manager; member
|
| D | graph_mgr.h | 94 CancellationManager* cancellation_manager,
171 CancellationManager* cancellation_manager,
|
| D | graph_mgr.cc | 412 CancellationManager* cancellation_manager, in ExecuteAsync() argument
479 cancellation_manager, session, in ExecuteAsync()
500 CostGraphDef* cost_graph, CancellationManager* cancellation_manager, in StartParallelExecutors() argument
520 args.cancellation_manager = cancellation_manager; in StartParallelExecutors()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 208 CancellationManager* cancellation_manager, in Run() argument
210 Status s = this->Run(&step_container_, inputs, outputs, cancellation_manager, in Run()
218 CancellationManager* cancellation_manager, in Run() argument
220 Status s = this->Run(&step_container_, inputs, outputs, cancellation_manager, in Run()
233 CancellationManager cancellation_manager; member
239 std::vector<Tensor>* outputs, CancellationManager* cancellation_manager, in Run() argument
256 if (cancellation_manager) { in Run()
257 params.cancellation_manager = cancellation_manager; in Run()
260 params.cancellation_manager = &op_execution_state->cancellation_manager; in Run()
268 params.cancellation_manager = &default_cancellation_manager; in Run()
[all …]
|
| D | kernel_and_device.h | 123 CancellationManager* cancellation_manager,
128 std::vector<Tensor>* outputs, CancellationManager* cancellation_manager,
189 CancellationManager* cancellation_manager,
195 CancellationManager* cancellation_manager,
269 CancellationManager* cancellation_manager,
274 CancellationManager* cancellation_manager,
|
| D | execute_node.h | 79 CancellationManager* cancellation_manager, bool async, in ExecuteNode() argument
87 cancellation_manager_(cancellation_manager), in ExecuteNode()
|
| D | eager_operation.h | 102 void SetCancellationManager(CancellationManager* cancellation_manager) { in SetCancellationManager() argument
103 cancellation_manager_ = cancellation_manager; in SetCancellationManager()
|
| D | execute.h | 54 GraphCollector* graph_collector, CancellationManager* cancellation_manager,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator_base.cc | 72 CancellationManager* cm = ctx->cancellation_manager(); in TryTakeGrad()
111 CancellationManager* cancellation_manager, CancellationToken token) { in Cancel() argument
117 if (attempt.cancellation_manager == cancellation_manager && in Cancel()
156 cur_attempt->context->cancellation_manager()); in TryAttemptLocked()
|
| D | conditional_accumulator_base.h | 96 CancellationManager* cancellation_manager; // not owned member
102 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
107 cancellation_manager(cancellation_manager), in Attempt()
138 void Cancel(CancellationManager* cancellation_manager,
|
| D | queue_base.h | 121 void Cancel(Action action, CancellationManager* cancellation_manager,
158 CancellationManager* cancellation_manager; // not owned member
167 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
172 cancellation_manager(cancellation_manager), in Attempt()
|
| D | queue_base.cc | 188 void QueueBase::Cancel(Action action, CancellationManager* cancellation_manager, in Cancel() argument
197 if (attempt.cancellation_manager == cancellation_manager && in Cancel()
305 cur_attempt->context->cancellation_manager()); in TryAttemptLocked()
|
| D | fifo_queue.cc | 52 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueue()
110 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueueMany()
156 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeue()
247 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
|
| D | priority_queue.cc | 73 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueue()
138 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueueMany()
199 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeue()
290 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 129 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueue()
187 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueueMany()
234 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeue()
326 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 85 params.cancellation_manager = &captured_state->cancellation_manager; in GetNext()
88 ctx->cancellation_manager(), in GetNext()
89 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in GetNext()
146 params.cancellation_manager = &new_state->cancellation_manager; in Restore()
149 ctx->cancellation_manager(), in Restore()
150 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in Restore()
180 params.cancellation_manager = &new_state->cancellation_manager; in SetIteratorFromDataset()
183 ctx->cancellation_manager(), in SetIteratorFromDataset()
184 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in SetIteratorFromDataset()
552 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
[all …]
|
| D | captured_function.cc | 653 CancellationManager cancellation_manager(ctx->cancellation_manager()); in Run() local
654 f_opts.cancellation_manager = &cancellation_manager; in Run()
691 CancellationManager cancellation_manager(ctx->cancellation_manager()); in RunWithBorrowedArgs() local
692 f_opts.cancellation_manager = &cancellation_manager; in RunWithBorrowedArgs()
730 CancellationManager cancellation_manager; in RunInstantiated() local
731 f_opts.cancellation_manager = &cancellation_manager; in RunInstantiated()
783 auto cancellation_manager = in RunAsync() local
784 absl::make_unique<CancellationManager>(ctx->cancellation_manager()); in RunAsync()
785 f_opts.cancellation_manager = cancellation_manager.get(); in RunAsync()
795 cancellation_manager.release(); in RunAsync()
|
| D | multi_device_iterator_ops.cc | 111 params.cancellation_manager = &cancellation_manager_; in GetNextFromShard()
114 ctx->cancellation_manager(), in GetNextFromShard()
115 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in GetNextFromShard()
147 CancellationManager* cancellation_manager() { return &cancellation_manager_; } in cancellation_manager() function in tensorflow::data::__anon1c56f74c0111::MultiDeviceIterator
565 params.cancellation_manager = resource->cancellation_manager(); in Compute()
569 ctx->cancellation_manager(), in Compute()
570 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | to_tf_record_op.cc | 79 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
80 params.cancellation_manager = &cancellation_manager; in DoCompute()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental.cc | 575 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerStartCancel() argument
576 cancellation_manager->cancellation_manager.StartCancel(); in TFE_CancellationManagerStartCancel()
580 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerIsCancelled() argument
581 return cancellation_manager->cancellation_manager.IsCancelled(); in TFE_CancellationManagerIsCancelled()
585 TFE_CancellationManager* cancellation_manager) { in TFE_DeleteCancellationManager() argument
586 delete cancellation_manager; in TFE_DeleteCancellationManager()
590 TFE_CancellationManager* cancellation_manager, in TFE_OpSetCancellationManager() argument
593 &cancellation_manager->cancellation_manager); in TFE_OpSetCancellationManager()
|
| /external/tensorflow/tensorflow/core/util/rpc/ |
| D | call_container.h | 105 token_(ctx->cancellation_manager() != nullptr in CallContainer()
106 ? ctx->cancellation_manager()->get_cancellation_token() in CallContainer()
116 ctx_->cancellation_manager()->DeregisterCallback(token_); in CallContainer()
134 is_cancelled = !ctx_->cancellation_manager()->RegisterCallback( in CallContainer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_runner.cc | 187 CancellationManager cancellation_manager; in Run() local
188 args.cancellation_manager = &cancellation_manager; in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous_test.cc | 163 args.cancellation_manager = cm; in TEST_F()
183 args.cancellation_manager = cm; in TEST_F()
205 args.cancellation_manager = cm; in TEST_F()
224 args_with_cancellation.cancellation_manager = cm; in TEST_F()
504 args.cancellation_manager = cm; in BM_PingPong()
|