Home
last modified time | relevance | path

Searched refs:OpKernelConstruction (Results 1 – 25 of 634) sorted by relevance

12345678910>>...26

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_ops.cc44 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 …]
Dvariable_ops.cc32 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 …]
Dreduction_ops.cc32 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()
Drelu_op.cc30 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()
Dshape_op.cc34 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()
Dpooling_ops.cc41 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 …]
Dsendrecv_ops.cc31 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()
Dindex_ops.cc33 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()
Dsegment_reduction_ops.cc29 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()
Dfft_ops.cc42 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()
Dtensor_list_ops.cc47 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()
Delu_op.cc31 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/
Dqueue_op.h36 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 …]
Dtranspose_op.h26 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()
Dcontrol_flow_ops.h29 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);
Dstack.h31 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);
Dqueue_op.cc28 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 …]
Dsummary_kernels.cc33 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 …]
Dnccl_ops.cc41 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()
Dsave_op.cc31 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()
Dresource_variable_ops.h24 explicit ReadVariableOp(OpKernelConstruction* c);
33 explicit ReadVariablesOp(OpKernelConstruction* c);
43 explicit DestroyResourceOp(OpKernelConstruction* ctx);
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h33 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/
Dmemory_stats_ops.cc22 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/
Dxla_ops.h104 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/
Dresource_ops.cc27 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()

12345678910>>...26