Home
last modified time | relevance | path

Searched refs:XlaOpKernel (Results 1 – 25 of 141) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drelu_op.cc38 class ReluOp : public XlaOpKernel {
40 explicit ReluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReluOp()
48 class Relu6Op : public XlaOpKernel {
50 explicit Relu6Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in Relu6Op()
58 class LeakyReluOp : public XlaOpKernel {
60 explicit LeakyReluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in LeakyReluOp()
74 class ReluGradOp : public XlaOpKernel {
76 explicit ReluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReluGradOp()
90 class Relu6GradOp : public XlaOpKernel {
92 explicit Relu6GradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in Relu6GradOp()
[all …]
Dvariable_ops.cc31 class VarIsInitializedOp : public XlaOpKernel {
33 explicit VarIsInitializedOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in VarIsInitializedOp()
43 class VariableShapeOp : public XlaOpKernel {
45 explicit VariableShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in VariableShapeOp()
64 class ReadVariableOp : public XlaOpKernel {
66 explicit ReadVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ReadVariableOp()
82 class AssignVariableOp : public XlaOpKernel {
84 explicit AssignVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignVariableOp()
92 class AssignAddVariableOp : public XlaOpKernel {
94 explicit AssignAddVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignAddVariableOp()
[all …]
Dshape_op.cc38 class ShapeOp : public XlaOpKernel {
40 explicit ShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeOp()
71 class XlaSetBoundOp : public XlaOpKernel {
74 : XlaOpKernel(context) {} in XlaSetBoundOp()
113 class XlaSetDynamicDimensionSizeOp : public XlaOpKernel {
116 : XlaOpKernel(context) {} in XlaSetDynamicDimensionSizeOp()
145 class ShapeNOp : public XlaOpKernel {
147 explicit ShapeNOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeNOp()
185 class RankOp : public XlaOpKernel {
187 explicit RankOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RankOp()
[all …]
Dscatter_nd_op.cc79 class ScatterNdOp : public XlaOpKernel {
81 explicit ScatterNdOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in ScatterNdOp()
172 class TensorScatterAddOp : public XlaOpKernel {
175 : XlaOpKernel(context) {} in TensorScatterAddOp()
185 class TensorScatterMaxOp : public XlaOpKernel {
188 : XlaOpKernel(context) {} in TensorScatterMaxOp()
198 class TensorScatterMinOp : public XlaOpKernel {
201 : XlaOpKernel(context) {} in TensorScatterMinOp()
211 class TensorScatterSubOp : public XlaOpKernel {
214 : XlaOpKernel(context) {} in TensorScatterSubOp()
[all …]
Delu_op.cc48 class EluOp : public XlaOpKernel {
50 explicit EluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluOp()
57 class EluGradOp : public XlaOpKernel {
59 explicit EluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluGradOp()
77 class SeluOp : public XlaOpKernel {
79 explicit SeluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SeluOp()
86 class SeluGradOp : public XlaOpKernel {
88 explicit SeluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SeluGradOp()
Dtensor_list_ops.cc87 class TensorListLengthOp : public XlaOpKernel {
89 explicit TensorListLengthOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TensorListLengthOp()
134 class TensorListReserveOp : public XlaOpKernel {
136 explicit TensorListReserveOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListReserveOp()
209 class EmptyTensorListOp : public XlaOpKernel {
211 explicit EmptyTensorListOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in EmptyTensorListOp()
282 class TensorListElementShapeOp : public XlaOpKernel {
285 : XlaOpKernel(ctx) { in TensorListElementShapeOp()
339 class TensorListGetItemOp : public XlaOpKernel {
341 explicit TensorListGetItemOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListGetItemOp()
[all …]
Drandom_ops.cc40 class RandomUniformOp : public XlaOpKernel {
42 explicit RandomUniformOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RandomUniformOp()
70 class RandomShuffleOp : public XlaOpKernel {
72 explicit RandomShuffleOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RandomShuffleOp()
219 class RandomUniformIntOp : public XlaOpKernel {
221 explicit RandomUniformIntOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RandomUniformIntOp()
255 class RandomStandardNormalOp : public XlaOpKernel {
258 : XlaOpKernel(ctx) {} in RandomStandardNormalOp()
284 class TruncatedNormalOp : public XlaOpKernel {
286 explicit TruncatedNormalOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TruncatedNormalOp()
[all …]
Dtensor_array_ops.cc133 class TensorArrayOp : public XlaOpKernel {
135 explicit TensorArrayOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorArrayOp()
191 class TensorArrayWriteOp : public XlaOpKernel {
193 explicit TensorArrayWriteOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorArrayWriteOp()
245 class TensorArrayReadOp : public XlaOpKernel {
247 explicit TensorArrayReadOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorArrayReadOp()
288 class TensorArrayGatherOp : public XlaOpKernel {
290 explicit TensorArrayGatherOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorArrayGatherOp()
355 class TensorArrayScatterOp : public XlaOpKernel {
357 explicit TensorArrayScatterOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorArrayScatterOp()
[all …]
Dstack_ops.cc91 class StackOp : public XlaOpKernel {
93 explicit StackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in StackOp()
127 class StackPushOp : public XlaOpKernel {
129 explicit StackPushOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in StackPushOp()
174 class StackPopOp : public XlaOpKernel {
176 explicit StackPopOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in StackPopOp()
228 class StackCloseOp : public XlaOpKernel {
230 explicit StackCloseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in StackCloseOp()
Dxla_self_adjoint_eig_op.cc24 class XlaSelfAdjointEigOp : public XlaOpKernel {
26 explicit XlaSelfAdjointEigOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in XlaSelfAdjointEigOp()
44 class SelfAdjointEigV2Op : public XlaOpKernel {
46 explicit SelfAdjointEigV2Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SelfAdjointEigV2Op()
Dsort_ops.cc26 class XlaSortOp : public XlaOpKernel {
28 explicit XlaSortOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in XlaSortOp()
40 class XlaKeyValueSortOp : public XlaOpKernel {
43 : XlaOpKernel(context) {} in XlaKeyValueSortOp()
58 class XlaVariadicSortOp : public XlaOpKernel {
61 : XlaOpKernel(context) { in XlaVariadicSortOp()
Dsendrecv_ops.cc29 class SendOp : public XlaOpKernel {
40 SendOp::SendOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SendOp()
53 class RecvOp : public XlaOpKernel {
65 RecvOp::RecvOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in RecvOp()
Dstateless_random_ops_v2.cc195 class StatelessRandomUniformOp : public XlaOpKernel {
198 : XlaOpKernel(ctx) { in StatelessRandomUniformOp()
250 class StatelessRandomUniformIntOp : public XlaOpKernel {
253 : XlaOpKernel(ctx) { in StatelessRandomUniformIntOp()
311 class StatelessRandomUniformFullIntOp : public XlaOpKernel {
314 : XlaOpKernel(ctx) { in StatelessRandomUniformFullIntOp()
357 class StatelessRandomNormalOp : public XlaOpKernel {
360 : XlaOpKernel(ctx) { in StatelessRandomNormalOp()
411 class StatelessTruncatedNormalOp : public XlaOpKernel {
414 : XlaOpKernel(ctx) { in StatelessTruncatedNormalOp()
[all …]
Dconv_ops.cc43 class ConvOp : public XlaOpKernel {
47 : XlaOpKernel(ctx) { in ConvOp()
94 class ConvBackpropInputOp : public XlaOpKernel {
98 : XlaOpKernel(ctx) { in ConvBackpropInputOp()
160 class ConvBackpropFilterOp : public XlaOpKernel {
164 : XlaOpKernel(ctx) { in ConvBackpropFilterOp()
Deinsum_op.cc33 class XlaEinsumOp : public XlaOpKernel {
35 explicit XlaEinsumOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in XlaEinsumOp()
57 class EinsumOp : public XlaOpKernel {
59 explicit EinsumOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in EinsumOp()
Dlower_upper_bound_ops.cc83 class LowerBoundOp : public XlaOpKernel {
85 explicit LowerBoundOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in LowerBoundOp()
99 class UpperBoundOp : public XlaOpKernel {
101 explicit UpperBoundOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in UpperBoundOp()
Dimage_resize_ops.h23 class ResizeNearestNeighborOp : public XlaOpKernel {
34 class ResizeBilinearOp : public XlaOpKernel {
46 class ResizeBilinearGradOp : public XlaOpKernel {
Dtraining_ops.cc29 class ResourceApplyGradientDescent : public XlaOpKernel {
32 : XlaOpKernel(ctx) {} in ResourceApplyGradientDescent()
72 class ResourceApplyProximalGradientDescent : public XlaOpKernel {
75 : XlaOpKernel(ctx) { in ResourceApplyProximalGradientDescent()
117 class ResourceApplyMomentum : public XlaOpKernel {
119 explicit ResourceApplyMomentum(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ResourceApplyMomentum()
174 class ResourceApplyKerasMomentum : public XlaOpKernel {
177 : XlaOpKernel(ctx) { in ResourceApplyKerasMomentum()
233 class ResourceApplyAdagrad : public XlaOpKernel {
235 explicit ResourceApplyAdagrad(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ResourceApplyAdagrad()
[all …]
Dgather_scatter_ops.cc25 class GatherOp : public XlaOpKernel {
27 explicit GatherOp(OpKernelConstruction* context) : XlaOpKernel(context) { in GatherOp()
55 class ScatterOp : public XlaOpKernel {
57 explicit ScatterOp(OpKernelConstruction* context) : XlaOpKernel(context) { in ScatterOp()
Dstateless_random_ops.cc141 class StatelessRandomUniformOp : public XlaOpKernel {
144 : XlaOpKernel(ctx), in StatelessRandomUniformOp()
188 class StatelessRandomUniformIntOp : public XlaOpKernel {
191 : XlaOpKernel(ctx), in StatelessRandomUniformIntOp()
239 class StatelessRandomUniformFullIntOp : public XlaOpKernel {
242 : XlaOpKernel(ctx), in StatelessRandomUniformFullIntOp()
280 class StatelessRandomNormalOp : public XlaOpKernel {
283 : XlaOpKernel(ctx), in StatelessRandomNormalOp()
331 class StatelessTruncatedNormalOp : public XlaOpKernel {
334 : XlaOpKernel(ctx), in StatelessTruncatedNormalOp()
Dxla_svd_op.cc25 class XlaSvdOp : public XlaOpKernel {
27 explicit XlaSvdOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in XlaSvdOp()
55 class SvdOp : public XlaOpKernel {
57 explicit SvdOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SvdOp()
Dreverse_op.cc31 class ReverseOp : public XlaOpKernel {
33 explicit ReverseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReverseOp()
69 class ReverseV2Op : public XlaOpKernel {
71 explicit ReverseV2Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReverseV2Op()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry_test.cc27 class DummyCPUOp : public XlaOpKernel {
29 explicit DummyCPUOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DummyCPUOp()
36 class DummyGenericOp : public XlaOpKernel {
38 explicit DummyGenericOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DummyGenericOp()
86 class DummyInfeasibleTypeConstraintOp : public XlaOpKernel {
89 : XlaOpKernel(ctx) {} in DummyInfeasibleTypeConstraintOp()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dcross_replica_ops.cc51 class CrossReplicaSumOp : public XlaOpKernel {
53 explicit CrossReplicaSumOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in CrossReplicaSumOp()
68 class AllToAllOp : public XlaOpKernel {
70 explicit AllToAllOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in AllToAllOp()
96 class CollectivePermuteOp : public XlaOpKernel {
98 explicit CollectivePermuteOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in CollectivePermuteOp()
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Doutfeed_ops.cc30 class OutfeedEnqueueOp : public XlaOpKernel {
32 explicit OutfeedEnqueueOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in OutfeedEnqueueOp()
54 class OutfeedEnqueueTupleOp : public XlaOpKernel {
56 explicit OutfeedEnqueueTupleOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in OutfeedEnqueueTupleOp()

123456