| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | relu_op.cc | 28 class ReluOp : public XlaOpKernel {
30 explicit ReluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReluOp()
40 class Relu6Op : public XlaOpKernel {
42 explicit Relu6Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in Relu6Op()
53 class LeakyReluOp : public XlaOpKernel {
55 explicit LeakyReluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in LeakyReluOp()
68 class ReluGradOp : public XlaOpKernel {
70 explicit ReluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ReluGradOp()
84 class Relu6GradOp : public XlaOpKernel {
86 explicit Relu6GradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in Relu6GradOp()
[all …]
|
| D | shape_op.cc | 32 class ShapeOp : public XlaOpKernel {
34 explicit ShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeOp()
51 class ShapeNOp : public XlaOpKernel {
53 explicit ShapeNOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeNOp()
73 class RankOp : public XlaOpKernel {
75 explicit RankOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RankOp()
89 class SizeOp : public XlaOpKernel {
91 explicit SizeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SizeOp()
114 class ExpandDimsOp : public XlaOpKernel {
116 explicit ExpandDimsOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ExpandDimsOp()
[all …]
|
| D | variable_ops.cc | 30 class VarIsInitializedOp : public XlaOpKernel {
32 explicit VarIsInitializedOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in VarIsInitializedOp()
42 class VariableShapeOp : public XlaOpKernel {
44 explicit VariableShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in VariableShapeOp()
63 class ReadVariableOp : public XlaOpKernel {
65 explicit ReadVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ReadVariableOp()
81 class AssignVariableOp : public XlaOpKernel {
83 explicit AssignVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignVariableOp()
91 class AssignAddVariableOp : public XlaOpKernel {
93 explicit AssignAddVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignAddVariableOp()
[all …]
|
| D | tensor_list_ops.cc | 45 class TensorListLengthOp : public XlaOpKernel {
47 explicit TensorListLengthOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TensorListLengthOp()
98 class TensorListReserveOp : public XlaOpKernel {
100 explicit TensorListReserveOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListReserveOp()
130 class EmptyTensorListOp : public XlaOpKernel {
132 explicit EmptyTensorListOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in EmptyTensorListOp()
166 class TensorListElementShapeOp : public XlaOpKernel {
169 : XlaOpKernel(ctx) { in TensorListElementShapeOp()
206 class TensorListGetItemOp : public XlaOpKernel {
208 explicit TensorListGetItemOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListGetItemOp()
[all …]
|
| D | elu_op.cc | 29 class EluOp : public XlaOpKernel {
31 explicit EluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluOp()
42 class EluGradOp : public XlaOpKernel {
44 explicit EluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluGradOp()
62 class SeluOp : public XlaOpKernel {
64 explicit SeluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SeluOp()
80 class SeluGradOp : public XlaOpKernel {
82 explicit SeluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SeluGradOp()
|
| D | tensor_array_ops.cc | 133 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 …]
|
| D | random_ops.cc | 38 class RandomUniformOp : public XlaOpKernel {
40 explicit RandomUniformOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RandomUniformOp()
64 class RandomShuffleOp : public XlaOpKernel {
66 explicit RandomShuffleOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RandomShuffleOp()
208 class RandomUniformIntOp : public XlaOpKernel {
210 explicit RandomUniformIntOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RandomUniformIntOp()
240 class RandomStandardNormalOp : public XlaOpKernel {
243 : XlaOpKernel(ctx) {} in RandomStandardNormalOp()
269 class TruncatedNormalOp : public XlaOpKernel {
271 explicit TruncatedNormalOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TruncatedNormalOp()
|
| D | diag_op.cc | 79 class DiagOp : public XlaOpKernel {
81 explicit DiagOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DiagOp()
121 class DiagPartOp : public XlaOpKernel {
123 explicit DiagPartOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DiagPartOp()
163 class MatrixDiagOp : public XlaOpKernel {
165 explicit MatrixDiagOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in MatrixDiagOp()
192 class MatrixDiagPartOp : public XlaOpKernel {
194 explicit MatrixDiagPartOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in MatrixDiagPartOp()
|
| D | stack_ops.cc | 91 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()
|
| D | sort_ops.cc | 23 class XlaSortOp : public XlaOpKernel {
25 explicit XlaSortOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in XlaSortOp()
34 class XlaKeyValueSortOp : public XlaOpKernel {
37 : XlaOpKernel(context) {} in XlaKeyValueSortOp()
|
| D | xla_self_adjoint_eig_op.cc | 24 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()
|
| D | conv_ops.cc | 42 class ConvOp : public XlaOpKernel {
46 : XlaOpKernel(ctx) { in ConvOp()
89 class ConvBackpropInputOp : public XlaOpKernel {
93 : XlaOpKernel(ctx) { in ConvBackpropInputOp()
147 class ConvBackpropFilterOp : public XlaOpKernel {
151 : XlaOpKernel(ctx) { in ConvBackpropFilterOp()
|
| D | stateless_random_ops.cc | 68 class StatelessRandomUniformOp : public XlaOpKernel {
71 : XlaOpKernel(ctx) { in StatelessRandomUniformOp()
109 class StatelessRandomUniformIntOp : public XlaOpKernel {
112 : XlaOpKernel(ctx) { in StatelessRandomUniformIntOp()
157 class StatelessRandomNormalOp : public XlaOpKernel {
160 : XlaOpKernel(ctx) { in StatelessRandomNormalOp()
198 class StatelessTruncatedNormalOp : public XlaOpKernel {
201 : XlaOpKernel(ctx) { in StatelessTruncatedNormalOp()
|
| D | sendrecv_ops.cc | 29 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()
|
| D | training_ops.cc | 29 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 …]
|
| D | dynamic_slice_ops.cc | 31 class DynamicUpdateSliceOp : public XlaOpKernel {
34 : XlaOpKernel(context) {} in DynamicUpdateSliceOp()
65 class DynamicSliceOp : public XlaOpKernel {
68 : XlaOpKernel(context) {} in DynamicSliceOp()
|
| D | xla_svd_op.cc | 25 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()
|
| D | reverse_op.cc | 31 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()
|
| D | bias_ops.cc | 29 class BiasOp : public XlaOpKernel {
31 explicit BiasOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BiasOp()
78 class BiasAddGradOp : public XlaOpKernel {
80 explicit BiasAddGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BiasAddGradOp()
|
| D | fake_quantize_ops.cc | 85 class FakeQuantWithMinMaxArgsOp : public XlaOpKernel {
88 : XlaOpKernel(ctx) { in FakeQuantWithMinMaxArgsOp()
133 class FakeQuantWithMinMaxArgsGradOp : public XlaOpKernel {
136 : XlaOpKernel(ctx) { in FakeQuantWithMinMaxArgsGradOp()
183 class FakeQuantWithMinMaxVarsOp : public XlaOpKernel {
186 : XlaOpKernel(ctx) { in FakeQuantWithMinMaxVarsOp()
222 class FakeQuantWithMinMaxVarsGradOp : public XlaOpKernel {
225 : XlaOpKernel(ctx) { in FakeQuantWithMinMaxVarsGradOp()
|
| D | softmax_op.cc | 33 class SoftmaxOp : public XlaOpKernel {
35 explicit SoftmaxOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SoftmaxOp()
142 class SoftmaxXentWithLogitsOp : public XlaOpKernel {
145 : XlaOpKernel(ctx) {} in SoftmaxXentWithLogitsOp()
175 class SparseSoftmaxXentWithLogitsOp : public XlaOpKernel {
178 : XlaOpKernel(ctx) {} in SparseSoftmaxXentWithLogitsOp()
|
| D | bcast_ops.cc | 32 class BCastArgsOp : public XlaOpKernel {
34 explicit BCastArgsOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BCastArgsOp()
79 class BCastGradArgsOp : public XlaOpKernel {
81 explicit BCastGradArgsOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BCastGradArgsOp()
|
| D | cast_op.cc | 29 class CastOp : public XlaOpKernel {
31 explicit CastOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in CastOp()
100 class BitcastOp : public XlaOpKernel {
102 explicit BitcastOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BitcastOp()
|
| D | transpose_op.cc | 34 class TransposeOp : public XlaOpKernel {
37 : XlaOpKernel(ctx), conjugate_(conjugate) {} in TransposeOp()
120 class InvertPermutationOp : public XlaOpKernel {
122 explicit InvertPermutationOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in InvertPermutationOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry_test.cc | 27 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()
|