/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | conv_ops.cc | 44 explicit ConvOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvOp() 69 explicit Conv2DOp(OpKernelConstruction* ctx) in Conv2DOp() 76 explicit Conv3DOp(OpKernelConstruction* ctx) in Conv3DOp() 83 explicit DepthwiseConv2DOp(OpKernelConstruction* ctx) in DepthwiseConv2DOp() 91 explicit ConvBackpropInputOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropInputOp() 122 explicit Conv2DBackpropInputOp(OpKernelConstruction* ctx) in Conv2DBackpropInputOp() 131 explicit Conv3DBackpropInputOp(OpKernelConstruction* ctx) in Conv3DBackpropInputOp() 140 explicit DepthwiseConv2DBackpropInputOp(OpKernelConstruction* ctx) in DepthwiseConv2DBackpropInputOp() 149 explicit ConvBackpropFilterOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropFilterOp() 180 explicit Conv2DBackpropFilterOp(OpKernelConstruction* ctx) in Conv2DBackpropFilterOp() [all …]
|
D | variable_ops.cc | 32 explicit VarIsInitializedOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in VarIsInitializedOp() 44 explicit VariableShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in VariableShapeOp() 65 explicit ReadVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ReadVariableOp() 83 explicit AssignVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignVariableOp() 93 explicit AssignAddVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignAddVariableOp() 109 explicit AssignSubVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignSubVariableOp() 125 explicit ResourceGatherOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ResourceGatherOp() 152 OpKernelConstruction* context, bool indices_are_vectors, in ResourceScatterOp() 188 explicit ResourceScatterAddOp(OpKernelConstruction* context) in ResourceScatterAddOp() 201 explicit ResourceScatterSubOp(OpKernelConstruction* context) in ResourceScatterSubOp() [all …]
|
D | reduction_ops.cc | 32 explicit SumOp(OpKernelConstruction* ctx) in SumOp() 49 explicit ProdOp(OpKernelConstruction* ctx) in ProdOp() 68 explicit MinOp(OpKernelConstruction* ctx) in MinOp() 86 explicit MaxOp(OpKernelConstruction* ctx) in MaxOp() 104 explicit MeanOp(OpKernelConstruction* ctx) in MeanOp() 139 explicit AllOp(OpKernelConstruction* ctx) in AllOp() 157 explicit AnyOp(OpKernelConstruction* ctx) in AnyOp()
|
D | relu_op.cc | 30 explicit ReluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReluOp() 42 explicit Relu6Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in Relu6Op() 55 explicit LeakyReluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in LeakyReluOp() 70 explicit ReluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReluGradOp() 86 explicit Relu6GradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in Relu6GradOp() 106 explicit LeakyReluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in LeakyReluGradOp()
|
D | shape_op.cc | 34 explicit ShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeOp() 53 explicit ShapeNOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeNOp() 75 explicit RankOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RankOp() 91 explicit SizeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SizeOp() 116 explicit ExpandDimsOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ExpandDimsOp() 162 explicit SqueezeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SqueezeOp() 223 explicit ZerosLikeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ZerosLikeOp() 237 explicit OnesLikeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in OnesLikeOp()
|
D | pooling_ops.cc | 41 PoolingOp(OpKernelConstruction* ctx, int num_spatial_dims, in PoolingOp() 152 MaxPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolOp() 185 explicit MaxPool2DOp(OpKernelConstruction* ctx) in MaxPool2DOp() 197 explicit MaxPool3DOp(OpKernelConstruction* ctx) in MaxPool3DOp() 204 AvgPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolOp() 247 explicit AvgPool2DOp(OpKernelConstruction* ctx) in AvgPool2DOp() 255 explicit AvgPool3DOp(OpKernelConstruction* ctx) in AvgPool3DOp() 268 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradOp() 354 explicit MaxPool2DGradOp(OpKernelConstruction* ctx) in MaxPool2DGradOp() 370 explicit MaxPool3DGradOp(OpKernelConstruction* ctx) in MaxPool3DGradOp() [all …]
|
D | sendrecv_ops.cc | 31 explicit SendOp(OpKernelConstruction* ctx); 40 SendOp::SendOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SendOp() 55 explicit RecvOp(OpKernelConstruction* ctx); 65 RecvOp::RecvOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in RecvOp()
|
D | index_ops.cc | 33 XlaArgMinMaxOp::XlaArgMinMaxOp(OpKernelConstruction* ctx, bool is_min) in XlaArgMinMaxOp() 76 XlaArgMaxOp::XlaArgMaxOp(OpKernelConstruction* ctx) in XlaArgMaxOp() 87 explicit XlaArgMinOp(OpKernelConstruction* ctx); 89 XlaArgMinOp::XlaArgMinOp(OpKernelConstruction* ctx) in XlaArgMinOp()
|
D | segment_reduction_ops.cc | 29 explicit UnsortedSegmentReduce(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in UnsortedSegmentReduce() 98 explicit UnsortedSegmentSum(OpKernelConstruction* ctx) in UnsortedSegmentSum() 113 explicit UnsortedSegmentProd(OpKernelConstruction* ctx) in UnsortedSegmentProd() 128 explicit UnsortedSegmentMin(OpKernelConstruction* ctx) in UnsortedSegmentMin() 145 explicit UnsortedSegmentMax(OpKernelConstruction* ctx) in UnsortedSegmentMax()
|
D | fft_ops.cc | 42 explicit GenericFftOp(OpKernelConstruction* ctx, FftType fft_type, in GenericFftOp() 113 explicit FFTOp(OpKernelConstruction* ctx) in FFTOp() 125 explicit IFFTOp(OpKernelConstruction* ctx) in IFFTOp() 138 explicit RFFTOp(OpKernelConstruction* ctx) in RFFTOp() 150 explicit IRFFTOp(OpKernelConstruction* ctx) in IRFFTOp()
|
D | tensor_list_ops.cc | 47 explicit TensorListLengthOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TensorListLengthOp() 100 explicit TensorListReserveOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListReserveOp() 132 explicit EmptyTensorListOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in EmptyTensorListOp() 168 explicit TensorListElementShapeOp(OpKernelConstruction* ctx) in TensorListElementShapeOp() 208 explicit TensorListGetItemOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListGetItemOp() 247 explicit TensorListStackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListStackOp() 267 explicit TensorListFromTensorOp(OpKernelConstruction* ctx) in TensorListFromTensorOp() 344 explicit TensorListSetItemOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListSetItemOp() 387 explicit TensorListPushBackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListPushBackOp() 431 explicit TensorListPopBackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListPopBackOp()
|
D | elu_op.cc | 31 explicit EluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluOp() 44 explicit EluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluGradOp() 64 explicit SeluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SeluOp() 82 explicit SeluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SeluGradOp()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | queue_op.h | 36 QueueOp(OpKernelConstruction* context); 68 explicit QueueOpKernel(OpKernelConstruction* context); 79 explicit QueueAccessOpKernel(OpKernelConstruction* context); 96 explicit EnqueueOp(OpKernelConstruction* context); 121 explicit EnqueueManyOp(OpKernelConstruction* context); 142 explicit DequeueOp(OpKernelConstruction* context); 167 explicit DequeueManyOp(OpKernelConstruction* context); 210 explicit DequeueUpToOp(OpKernelConstruction* context); 228 explicit QueueCloseOp(OpKernelConstruction* context); 247 explicit QueueSizeOp(OpKernelConstruction* context); [all …]
|
D | transpose_op.h | 26 explicit TransposeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in TransposeOp() 38 explicit TransposeCpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in TransposeCpuOp() 48 explicit MklTransposeCpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in MklTransposeCpuOp() 58 explicit TransposeGpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in TransposeGpuOp() 68 explicit TransposeSyclOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in TransposeSyclOp() 79 explicit ConjugateTransposeCpuOp(OpKernelConstruction* ctx) in ConjugateTransposeCpuOp() 91 explicit MklConjugateTransposeCpuOp(OpKernelConstruction* ctx) in MklConjugateTransposeCpuOp() 103 explicit ConjugateTransposeGpuOp(OpKernelConstruction* ctx) in ConjugateTransposeGpuOp() 115 explicit ConjugateTransposeSyclOp(OpKernelConstruction* ctx) in ConjugateTransposeSyclOp()
|
D | control_flow_ops.h | 29 explicit ControlTriggerOp(OpKernelConstruction* context) in ControlTriggerOp() 41 explicit SwitchOp(OpKernelConstruction* context) : OpKernel(context) {} in SwitchOp() 54 explicit MergeOp(OpKernelConstruction* context); 67 explicit EnterOp(OpKernelConstruction* context) : OpKernel(context) {} in EnterOp() 80 explicit ExitOp(OpKernelConstruction* context) : OpKernel(context) {} in ExitOp() 92 explicit NextIterationOp(OpKernelConstruction* context) : OpKernel(context) {} in NextIterationOp() 106 explicit LoopCondOp(OpKernelConstruction* context);
|
D | stack.h | 31 explicit StackOp(OpKernelConstruction* context); 43 StackPushOp(OpKernelConstruction* context, bool allow_swapping); 56 TemplatedStackPushOp(OpKernelConstruction* context) in TemplatedStackPushOp() 62 explicit StackPopOp(OpKernelConstruction* context); 69 explicit StackCloseOp(OpKernelConstruction* context);
|
D | queue_op.cc | 28 QueueOp::QueueOp(OpKernelConstruction* context) : ResourceOpKernel(context) { in QueueOp() 50 QueueOpKernel::QueueOpKernel(OpKernelConstruction* context) in QueueOpKernel() 68 QueueAccessOpKernel::QueueAccessOpKernel(OpKernelConstruction* context) in QueueAccessOpKernel() 85 EnqueueOp::EnqueueOp(OpKernelConstruction* context) in EnqueueOp() 126 EnqueueManyOp::EnqueueManyOp(OpKernelConstruction* context) in EnqueueManyOp() 163 DequeueOp::DequeueOp(OpKernelConstruction* context) in DequeueOp() 206 DequeueManyOp::DequeueManyOp(OpKernelConstruction* context) in DequeueManyOp() 279 DequeueUpToOp::DequeueUpToOp(OpKernelConstruction* context) in DequeueUpToOp() 327 QueueCloseOp::QueueCloseOp(OpKernelConstruction* context) in QueueCloseOp() 344 QueueSizeOp::QueueSizeOp(OpKernelConstruction* context) in QueueSizeOp() [all …]
|
D | summary_kernels.cc | 33 explicit CreateSummaryFileWriterOp(OpKernelConstruction* ctx) in CreateSummaryFileWriterOp() 64 explicit CreateSummaryDbWriterOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in CreateSummaryDbWriterOp() 101 explicit FlushSummaryWriterOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in FlushSummaryWriterOp() 115 explicit CloseSummaryWriterOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in CloseSummaryWriterOp() 127 explicit WriteSummaryOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in WriteSummaryOp() 152 explicit ImportEventOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ImportEventOp() 173 explicit WriteScalarSummaryOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in WriteScalarSummaryOp() 196 explicit WriteHistogramSummaryOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in WriteHistogramSummaryOp() 219 explicit WriteImageSummaryOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in WriteImageSummaryOp() 257 explicit WriteAudioSummaryOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in WriteAudioSummaryOp() [all …]
|
D | nccl_ops.cc | 41 explicit NcclAsyncOpBase(OpKernelConstruction* c) : AsyncOpKernel(c) { in NcclAsyncOpBase() 63 explicit NcclReduceOpBase(OpKernelConstruction* c) : NcclAsyncOpBase(c) { in NcclReduceOpBase() 90 explicit NcclAllReduceOpKernel(OpKernelConstruction* c) in NcclAllReduceOpKernel() 127 explicit NcclReduceSendKernel(OpKernelConstruction* c) in NcclReduceSendKernel() 159 explicit NcclReduceRecvKernel(OpKernelConstruction* c) in NcclReduceRecvKernel() 199 explicit NcclBroadcastSendKernel(OpKernelConstruction* c) in NcclBroadcastSendKernel() 229 explicit NcclBroadcastRecvKernel(OpKernelConstruction* c) in NcclBroadcastRecvKernel() 265 explicit NcclStubKernel(OpKernelConstruction* c) : AsyncOpKernel(c) {} in NcclStubKernel()
|
D | save_op.cc | 31 explicit SaveOp(OpKernelConstruction* context) : OpKernel(context) {} in SaveOp() 42 explicit SaveSlicesOp(OpKernelConstruction* context) : OpKernel(context) {} in SaveSlicesOp() 53 explicit ShardedFilenameOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShardedFilenameOp() 76 explicit ShardedFilespecOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShardedFilespecOp()
|
D | resource_variable_ops.h | 24 explicit ReadVariableOp(OpKernelConstruction* c); 33 explicit ReadVariablesOp(OpKernelConstruction* c); 43 explicit DestroyResourceOp(OpKernelConstruction* ctx);
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | iterator_ops.h | 33 explicit IteratorHandleOp(OpKernelConstruction* ctx); 84 explicit AnonymousIteratorHandleOp(OpKernelConstruction* context); 103 explicit MakeIteratorOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in MakeIteratorOp() 110 explicit IteratorGetNextOp(OpKernelConstruction* ctx) in IteratorGetNextOp() 122 explicit IteratorGetNextAsOptionalOp(OpKernelConstruction* ctx) in IteratorGetNextAsOptionalOp() 140 explicit IteratorGetNextSyncOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in IteratorGetNextSyncOp() 147 explicit IteratorToStringHandleOp(OpKernelConstruction* ctx) in IteratorToStringHandleOp() 155 explicit IteratorFromStringHandleOp(OpKernelConstruction* ctx);
|
/external/tensorflow/tensorflow/contrib/memory_stats/kernels/ |
D | memory_stats_ops.cc | 22 explicit MemoryStatsOp(OpKernelConstruction* context) : OpKernel(context) {} in MemoryStatsOp() 48 explicit BytesInUseOp(OpKernelConstruction* context) in BytesInUseOp() 70 explicit BytesLimitOp(OpKernelConstruction* context) in BytesLimitOp() 92 explicit MaxBytesInUseOp(OpKernelConstruction* context) in MaxBytesInUseOp()
|
/external/tensorflow/tensorflow/compiler/jit/kernels/ |
D | xla_ops.h | 104 XlaLocalLaunchBase(OpKernelConstruction* ctx, 135 explicit XlaLocalLaunchOp(OpKernelConstruction* ctx); 144 explicit XlaCompileOp(OpKernelConstruction* ctx); 171 explicit XlaRunOp(OpKernelConstruction* ctx);
|
/external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
D | resource_ops.cc | 27 explicit TensorForestCreateTreeVariableOp(OpKernelConstruction* context) in TensorForestCreateTreeVariableOp() 54 explicit TensorForestTreeSerializeOp(OpKernelConstruction* context) in TensorForestTreeSerializeOp() 74 explicit TensorForestTreeDeserializeOp(OpKernelConstruction* context) in TensorForestTreeDeserializeOp() 101 explicit TensorForestTreeSizeOp(OpKernelConstruction* context) in TensorForestTreeSizeOp()
|