| /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()
|