Home
last modified time | relevance | path

Searched refs:AsyncOpKernel (Results 1 – 25 of 69) sorted by relevance

123

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc25 class CopyFromGpuToHostKernel : public AsyncOpKernel {
28 : AsyncOpKernel(context) {} in CopyFromGpuToHostKernel()
58 class CopyFromHostToGpuKernel : public AsyncOpKernel {
61 : AsyncOpKernel(context) {} in CopyFromHostToGpuKernel()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc187 class BatchFunctionKernel : public AsyncOpKernel {
189 explicit BatchFunctionKernel(OpKernelConstruction* c) : AsyncOpKernel(c) { in BatchFunctionKernel()
459 class BatchKernel : public AsyncOpKernel {
461 explicit BatchKernel(OpKernelConstruction* c) : AsyncOpKernel(c) { in BatchKernel()
560 Status Compute(OpKernelContext* context, AsyncOpKernel::DoneCallback done) { in Compute()
599 std::vector<AsyncOpKernel::DoneCallback> done_callbacks_to_call; in Compute()
651 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute()
704 AsyncOpKernel::DoneCallback done;
722 class UnbatchKernel : public AsyncOpKernel {
724 explicit UnbatchKernel(OpKernelConstruction* c) : AsyncOpKernel(c) { in UnbatchKernel()
[all …]
Dstack.h41 class StackPushOp : public AsyncOpKernel {
60 class StackPopOp : public AsyncOpKernel {
Dnccl_ops.cc43 class NcclAsyncOpBase : public AsyncOpKernel {
45 explicit NcclAsyncOpBase(OpKernelConstruction* c) : AsyncOpKernel(c) { in NcclAsyncOpBase()
262 class NcclStubKernel : public AsyncOpKernel {
264 explicit NcclStubKernel(OpKernelConstruction* c) : AsyncOpKernel(c) {} in NcclStubKernel()
Dreader_ops.cc45 class ReaderVerbAsyncOpKernel : public AsyncOpKernel {
47 using AsyncOpKernel::AsyncOpKernel;
50 : AsyncOpKernel(context), in ReaderVerbAsyncOpKernel()
Dfunctional_ops.cc126 class IfOp : public AsyncOpKernel {
128 explicit IfOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in IfOp()
243 class CaseOp : public AsyncOpKernel {
245 explicit CaseOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in CaseOp()
356 class WhileOp : public AsyncOpKernel {
358 explicit WhileOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in WhileOp()
728 class ForOp : public AsyncOpKernel {
730 explicit ForOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in ForOp()
Dcollective_ops.cc53 class CollectiveOpV1Kernel : public AsyncOpKernel {
56 : AsyncOpKernel(c), name_(name()), col_params_(new CollectiveParams()) {} in CollectiveOpV1Kernel()
466 class CollectiveReduceV2OpKernel : public AsyncOpKernel {
469 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveReduceV2OpKernel()
616 class CollectiveGatherV2OpKernel : public AsyncOpKernel {
619 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveGatherV2OpKernel()
751 class CollectiveBcastSendV2OpKernel : public AsyncOpKernel {
754 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveBcastSendV2OpKernel()
877 class CollectiveBcastRecvV2OpKernel : public AsyncOpKernel {
880 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveBcastRecvV2OpKernel()
Dlookup_table_init_op.cc169 class InitializeTableFromDatasetOp : public AsyncOpKernel {
172 : AsyncOpKernel(ctx), in InitializeTableFromDatasetOp()
Dsendrecv_ops.h39 class RecvOp : public AsyncOpKernel {
Dmutex_ops.cc136 class MutexLockOp : public AsyncOpKernel {
138 explicit MutexLockOp(OpKernelConstruction* c) : AsyncOpKernel(c) {} in MutexLockOp()
Dconditional_accumulator_base_op.h191 class ConditionalAccumulatorBaseAsyncOpKernel : public AsyncOpKernel {
195 : AsyncOpKernel(context) {} in ConditionalAccumulatorBaseAsyncOpKernel()
Dlookup_util.h64 AsyncOpKernel::DoneCallback done);
Dpartitioned_function_ops.h35 class PartitionedCallOp : public AsyncOpKernel {
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.h29 class TPUExecuteOp : public AsyncOpKernel {
34 AsyncOpKernel* AsAsync() override;
/external/tensorflow/tensorflow/core/common_runtime/
Dtestlib_ops.cc86 class DelayOp : public AsyncOpKernel {
88 explicit DelayOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in DelayOp()
Drenamed_device.h126 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
127 AsyncOpKernel::DoneCallback done) override { in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.h51 AsyncOpKernel::DoneCallback done_callback);
72 AsyncOpKernel::DoneCallback done_callback;
99 int split_index, AsyncOpKernel::DoneCallback done_callback);
/external/tensorflow/tensorflow/core/framework/
Ddevice.h92 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
93 AsyncOpKernel::DoneCallback done) { in ComputeAsync()
Dqueue_interface.h34 typedef AsyncOpKernel::DoneCallback DoneCallback;
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc31 class ToTFRecordOp : public AsyncOpKernel {
34 : AsyncOpKernel(ctx), in ToTFRecordOp()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dself_adjoint_eig_v2_op_gpu.cc41 class SelfAdjointEigV2OpGpu : public AsyncOpKernel {
44 : AsyncOpKernel(context) { in SelfAdjointEigV2OpGpu()
Ddeterminant_op.cc128 class DeterminantOpGpu : public AsyncOpKernel {
131 : AsyncOpKernel(context) {} in DeterminantOpGpu()
268 class LogDeterminantOpGpu : public AsyncOpKernel {
271 : AsyncOpKernel(context) {} in LogDeterminantOpGpu()
Dcholesky_op.cc94 class CholeskyOpGpu : public AsyncOpKernel {
97 : AsyncOpKernel(context) {} in CholeskyOpGpu()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h147 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context,
148 AsyncOpKernel::DoneCallback done) override;
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.h35 class MapDefunOp : public AsyncOpKernel {

123