| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 25 class CopyFromGpuToHostKernel : public AsyncOpKernel {
28 : AsyncOpKernel(context) {} in CopyFromGpuToHostKernel()
58 class CopyFromHostToGpuKernel : public AsyncOpKernel {
61 : AsyncOpKernel(context) {} in CopyFromHostToGpuKernel()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 108 class IteratorGetNextOp : public AsyncOpKernel {
111 : AsyncOpKernel(ctx), in IteratorGetNextOp()
120 class IteratorGetNextAsOptionalOp : public AsyncOpKernel {
123 : AsyncOpKernel(ctx), in IteratorGetNextAsOptionalOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 242 AsyncOpKernel::DoneCallback done_callback) { in RegisterInput()
289 AsyncOpKernel::DoneCallback done_callback;
563 AsyncOpKernel::DoneCallback last_task_callback = in ProcessBatch()
691 class BatchFunctionKernel : public AsyncOpKernel {
693 explicit BatchFunctionKernel(OpKernelConstruction* c) : AsyncOpKernel(c) { in BatchFunctionKernel()
780 class BatchKernel : public AsyncOpKernel {
782 explicit BatchKernel(OpKernelConstruction* c) : AsyncOpKernel(c) { in BatchKernel()
881 Status Compute(OpKernelContext* context, AsyncOpKernel::DoneCallback done) { in Compute()
930 std::vector<AsyncOpKernel::DoneCallback> done_callbacks_to_call; in Compute()
982 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute()
[all …]
|
| D | reader_ops.cc | 45 class ReaderVerbAsyncOpKernel : public AsyncOpKernel {
47 using AsyncOpKernel::AsyncOpKernel;
50 : AsyncOpKernel(context), in ReaderVerbAsyncOpKernel()
|
| D | functional_ops.cc | 127 class IfOp : public AsyncOpKernel {
129 explicit IfOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in IfOp()
215 class CaseOp : public AsyncOpKernel {
217 explicit CaseOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in CaseOp()
323 class WhileOp : public AsyncOpKernel {
325 explicit WhileOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in WhileOp()
498 class ForOp : public AsyncOpKernel {
500 explicit ForOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in ForOp()
|
| D | stack.h | 41 class StackPushOp : public AsyncOpKernel {
60 class StackPopOp : public AsyncOpKernel {
|
| D | nccl_ops.cc | 39 class NcclAsyncOpBase : public AsyncOpKernel {
41 explicit NcclAsyncOpBase(OpKernelConstruction* c) : AsyncOpKernel(c) { in NcclAsyncOpBase()
263 class NcclStubKernel : public AsyncOpKernel {
265 explicit NcclStubKernel(OpKernelConstruction* c) : AsyncOpKernel(c) {} in NcclStubKernel()
|
| D | sendrecv_ops.cc | 123 RecvOp::RecvOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in RecvOp()
147 AsyncOpKernel::DoneCallback done) { in make_recv_callback()
150 [ctx](AsyncOpKernel::DoneCallback done, in make_recv_callback()
|
| D | rpc_op.cc | 39 class RpcOp : public AsyncOpKernel {
41 explicit RpcOp(OpKernelConstruction* context) : AsyncOpKernel(context) { in RpcOp()
|
| D | cholesky_op.cc | 95 class CholeskyOpGpu : public AsyncOpKernel {
98 : AsyncOpKernel(context) {} in CholeskyOpGpu()
|
| D | determinant_op.cc | 128 class DeterminantOpGpu : public AsyncOpKernel {
131 : AsyncOpKernel(context) {} in DeterminantOpGpu()
268 class LogDeterminantOpGpu : public AsyncOpKernel {
271 : AsyncOpKernel(context) {} in LogDeterminantOpGpu()
|
| D | self_adjoint_eig_v2_op_gpu.cc | 41 class SelfAdjointEigV2OpGpu : public AsyncOpKernel {
44 : AsyncOpKernel(context) { in SelfAdjointEigV2OpGpu()
|
| D | crop_and_resize_op.cc | 109 class CropAndResizeOp : public AsyncOpKernel {
112 : AsyncOpKernel(context) { in CropAndResizeOp()
334 class CropAndResizeGradImageOp : public AsyncOpKernel {
337 : AsyncOpKernel(context) { in CropAndResizeGradImageOp()
516 class CropAndResizeGradBoxesOp : public AsyncOpKernel {
519 : AsyncOpKernel(context) { in CropAndResizeGradBoxesOp()
|
| D | conditional_accumulator_base_op.h | 123 class ConditionalAccumulatorBaseAsyncOpKernel : public AsyncOpKernel {
127 : AsyncOpKernel(context) {} in ConditionalAccumulatorBaseAsyncOpKernel()
|
| D | check_numerics_op.cc | 115 class CheckNumericsOp<GPUDevice, T> : public AsyncOpKernel {
120 : AsyncOpKernel(context) { in CheckNumericsOp()
|
| D | mutex_ops.cc | 168 class MutexLockOp : public AsyncOpKernel {
170 explicit MutexLockOp(OpKernelConstruction* c) : AsyncOpKernel(c) {} in MutexLockOp()
|
| /external/tensorflow/tensorflow/core/util/rpc/ |
| D | call_container.h | 68 bool try_rpc, AsyncOpKernel::DoneCallback done,
87 const AsyncOpKernel::DoneCallback done_;
100 AsyncOpKernel::DoneCallback done, in CallContainer()
|
| D | rpc_factory.h | 63 AsyncOpKernel::DoneCallback done) = 0;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | testlib_ops.cc | 78 class DelayOp : public AsyncOpKernel {
80 explicit DelayOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in DelayOp()
|
| D | device.h | 93 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
94 AsyncOpKernel::DoneCallback done) { in ComputeAsync()
|
| D | renamed_device.h | 103 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
104 AsyncOpKernel::DoneCallback done) override { in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | to_tf_record_op.cc | 28 class ToTFRecordOp : public AsyncOpKernel {
31 : AsyncOpKernel(ctx), in ToTFRecordOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | function_ops.cc | 59 class SymbolicGradientOp : public AsyncOpKernel {
62 : AsyncOpKernel(ctx), handle_(-1) {} in SymbolicGradientOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 57 AsyncHelper(AsyncOpKernel::DoneCallback done) { done_ = done; } in AsyncHelper()
61 AsyncOpKernel::DoneCallback done_;
66 class TRTEngineOp : public AsyncOpKernel {
71 AsyncOpKernel::DoneCallback done) override;
187 : AsyncOpKernel(context) { in TRTEngineOp()
364 AsyncOpKernel::DoneCallback done) { in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.h | 142 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context,
143 AsyncOpKernel::DoneCallback done) override;
|