| /external/tensorflow/tensorflow/core/kernels/ |
| D | fifo_queue.cc | 54 bool already_cancelled; in TryEnqueue() local
57 already_cancelled = !cm->RegisterCallback( in TryEnqueue()
59 if (!already_cancelled) { in TryEnqueue()
79 if (!already_cancelled) { in TryEnqueue()
112 bool already_cancelled; in TryEnqueueMany() local
115 already_cancelled = !cm->RegisterCallback( in TryEnqueueMany()
117 if (!already_cancelled) { in TryEnqueueMany()
147 if (!already_cancelled) { in TryEnqueueMany()
158 bool already_cancelled; in TryDequeue() local
161 already_cancelled = !cm->RegisterCallback( in TryDequeue()
[all …]
|
| D | priority_queue.cc | 75 bool already_cancelled; in TryEnqueue() local
78 already_cancelled = !cm->RegisterCallback( in TryEnqueue()
80 if (!already_cancelled) { in TryEnqueue()
108 if (!already_cancelled) { in TryEnqueue()
140 bool already_cancelled; in TryEnqueueMany() local
143 already_cancelled = !cm->RegisterCallback( in TryEnqueueMany()
145 if (!already_cancelled) { in TryEnqueueMany()
190 if (!already_cancelled) { in TryEnqueueMany()
202 bool already_cancelled; in TryDequeue() local
205 already_cancelled = !cm->RegisterCallback( in TryDequeue()
[all …]
|
| D | random_shuffle_queue_op.cc | 132 bool already_cancelled; in TryEnqueue() local
135 already_cancelled = !cm->RegisterCallback( in TryEnqueue()
137 if (!already_cancelled) { in TryEnqueue()
157 if (!already_cancelled) { in TryEnqueue()
190 bool already_cancelled; in TryEnqueueMany() local
193 already_cancelled = !cm->RegisterCallback( in TryEnqueueMany()
195 if (!already_cancelled) { in TryEnqueueMany()
225 if (!already_cancelled) { in TryEnqueueMany()
237 bool already_cancelled; in TryDequeue() local
240 already_cancelled = !cm->RegisterCallback( in TryDequeue()
[all …]
|
| D | conditional_accumulator_base.cc | 74 bool already_cancelled; in TryTakeGrad() local
77 already_cancelled = !cm->RegisterCallback( in TryTakeGrad()
79 if (!already_cancelled) { in TryTakeGrad()
98 if (!already_cancelled) { in TryTakeGrad()
|
| D | padding_fifo_queue.cc | 91 bool already_cancelled; in TryDequeueMany() local
94 already_cancelled = !cm->RegisterCallback( in TryDequeueMany()
96 if (!already_cancelled) { in TryDequeueMany()
230 if (!already_cancelled) { in TryDequeueMany()
|
| D | mutex_ops.cc | 86 const bool already_cancelled = in AcquireAsync() local
92 if (already_cancelled) { in AcquireAsync()
|
| D | control_flow_ops.cc | 463 bool already_cancelled = cm->IsCancelled(); in Compute() local
464 OP_REQUIRES(context, !already_cancelled, in Compute()
|
| D | collective_ops.cc | 70 const bool already_cancelled = in ComputeAsync() local
77 OP_REQUIRES_ASYNC(c, !already_cancelled, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | transfer_ops.cc | 49 bool already_cancelled; in ComputeAsync() local
54 already_cancelled = !ctx->cancellation_manager()->RegisterCallback( in ComputeAsync()
57 OP_REQUIRES_ASYNC(ctx, !already_cancelled, in ComputeAsync()
|
| D | tpu_compile_op_common.cc | 546 const bool already_cancelled = in Compute() local
560 OP_REQUIRES(ctx, !already_cancelled, in Compute()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 352 bool already_cancelled = !cancellation_manager->RegisterCallback( in RegisterCancellation() local
355 return std::pair<CancellationToken, bool>(token, already_cancelled); in RegisterCancellation()
486 bool already_cancelled; in TPUExecute() local
487 std::tie(token, already_cancelled) = in TPUExecute()
493 if (already_cancelled) { in TPUExecute()
509 already_cancelled = cancellation_manager->IsCancelling() || in TPUExecute()
511 if (already_cancelled) { in TPUExecute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | reader_op_kernel.h | 57 bool already_cancelled = in Compute() local
60 if (!already_cancelled) { in Compute()
|
| D | local_rendezvous.cc | 188 bool already_cancelled = false; in RecvAsync() local
205 already_cancelled = !cm->RegisterCallback(token, [this, token, key_hash] { in RecvAsync()
251 if (already_cancelled) { in RecvAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.cc | 192 bool already_cancelled = false; in ConsumeBuf() local
195 already_cancelled = !cancellation_manager->RegisterCallback( in ConsumeBuf()
198 if (already_cancelled) { in ConsumeBuf()
|
| D | collective_param_resolver_local.cc | 105 const bool already_cancelled = in CompleteGroupLocal() local
113 if (already_cancelled) { in CompleteGroupLocal()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.cc | 57 const bool already_cancelled = !cm->RegisterCallback( in RunAsync() local
59 if (already_cancelled) { in RunAsync()
|
| D | cluster_function_library_runtime.cc | 209 const bool already_cancelled = !cm->RegisterCallback( in Run() local
212 if (already_cancelled) { in Run()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 89 bool already_cancelled = in Enqueue() local
94 if (already_cancelled) { in Enqueue()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 127 const bool already_cancelled = in Compute() local
142 OP_REQUIRES(ctx, !already_cancelled, in Compute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 404 bool already_cancelled = false; in RegisterCall() local
414 already_cancelled = !cm->RegisterCallback(token, [this, call] { in RegisterCall()
424 if (already_cancelled) { in RegisterCall()
|
| D | worker.cc | 213 bool already_cancelled = !cancellation_manager_.RegisterCallback( in DoRunGraph() local
215 if (already_cancelled) { in DoRunGraph()
|