Home
last modified time | relevance | path

Searched refs:REGISTER_XLA_OP (Results 1 – 25 of 149) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dunary_ops.cc47 REGISTER_XLA_OP(Name(#NAME), NAME##Op);
56 REGISTER_XLA_OP(Name("Abs"), MlirXlaOpKernel);
61 REGISTER_XLA_OP(Name("Atan"), MlirXlaOpKernel);
63 REGISTER_XLA_OP(Name("Ceil"), MlirXlaOpKernel);
64 REGISTER_XLA_OP(Name("Cos"), MlirXlaOpKernel);
67 REGISTER_XLA_OP(Name("Exp"), MlirXlaOpKernel);
68 REGISTER_XLA_OP(Name("Expm1"), MlirXlaOpKernel);
69 REGISTER_XLA_OP(Name("Floor"), MlirXlaOpKernel);
70 REGISTER_XLA_OP(Name("IsFinite"), MlirXlaOpKernel);
71 REGISTER_XLA_OP(Name("IsInf"), MlirXlaOpKernel);
[all …]
Didentity_op.cc48 REGISTER_XLA_OP(
51 REGISTER_XLA_OP(Name("IdentityN")
56 REGISTER_XLA_OP(Name("PlaceholderWithDefault"), IdentityOp);
57 REGISTER_XLA_OP(Name("PreventGradient"), MlirXlaOpKernel);
58 REGISTER_XLA_OP(Name("StopGradient").AllowVariantTypes(), IdentityOp);
59 REGISTER_XLA_OP(Name("Snapshot"), IdentityOp);
60 REGISTER_XLA_OP(Name("_EagerConst"), IdentityOp);
Dfft_ops.cc118 REGISTER_XLA_OP(Name("FFT").TypeConstraint("Tcomplex",
121 REGISTER_XLA_OP(Name("FFT2D").TypeConstraint("Tcomplex",
124 REGISTER_XLA_OP(Name("FFT3D").TypeConstraint("Tcomplex",
134 REGISTER_XLA_OP(Name("IFFT").TypeConstraint("Tcomplex", DT_COMPLEX64),
136 REGISTER_XLA_OP(Name("IFFT2D").TypeConstraint("Tcomplex", DT_COMPLEX64),
138 REGISTER_XLA_OP(Name("IFFT3D").TypeConstraint("Tcomplex", DT_COMPLEX64),
147 REGISTER_XLA_OP(Name("RFFT")
152 REGISTER_XLA_OP(Name("RFFT2D")
157 REGISTER_XLA_OP(Name("RFFT3D")
169 REGISTER_XLA_OP(Name("IRFFT")
[all …]
Dvariable_ops.cc54 REGISTER_XLA_OP(Name("VarIsInitializedOp"), VarIsInitializedOp);
75 REGISTER_XLA_OP(Name("VariableShape").CompilationOnly().IsMetadataOp(),
94 REGISTER_XLA_OP(Name("ReadVariableOp").CompilationOnly(), ReadVariableOp);
104 REGISTER_XLA_OP(Name("AssignVariableOp").CompilationOnly(), AssignVariableOp);
119 REGISTER_XLA_OP(
136 REGISTER_XLA_OP(
161 REGISTER_XLA_OP(Name("ResourceGather"), ResourceGatherOp);
217 REGISTER_XLA_OP(Name("ResourceScatterAdd"), ResourceScatterAddOp);
230 REGISTER_XLA_OP(Name("ResourceScatterSub"), ResourceScatterSubOp);
243 REGISTER_XLA_OP(Name("ResourceScatterMul"), ResourceScatterMulOp);
[all …]
Dconv_ops.cc73 REGISTER_XLA_OP(Name("Conv2D").TypeConstraint("T", GetXlaConvTypes()),
81 REGISTER_XLA_OP(Name("Conv3D").TypeConstraint("T", GetXlaConvTypes()),
89 REGISTER_XLA_OP(
137 REGISTER_XLA_OP(Name("Conv2DBackpropInput")
147 REGISTER_XLA_OP(Name("Conv3DBackpropInputV2")
157 REGISTER_XLA_OP(Name("DepthwiseConv2dNativeBackpropInput")
201 REGISTER_XLA_OP(Name("Conv2DBackpropFilter")
212 REGISTER_XLA_OP(Name("Conv3DBackpropFilterV2")
222 REGISTER_XLA_OP(Name("DepthwiseConv2dNativeBackpropFilter")
Dfunction_ops.cc51 REGISTER_XLA_OP(Name("_ListToArray"), PassOn);
52 REGISTER_XLA_OP(Name("_ArrayToList"), PassOn);
72 REGISTER_XLA_OP(Name(kGradientOp), AlwaysFailOp);
73 REGISTER_XLA_OP(Name("PartitionedCall")
78 REGISTER_XLA_OP(Name("StatefulPartitionedCall")
Drelu_op.cc39 REGISTER_XLA_OP(Name("Relu"), MlirXlaOpKernel);
40 REGISTER_XLA_OP(Name("Relu6"), MlirXlaOpKernel);
56 REGISTER_XLA_OP(Name("LeakyRelu"), LeakyReluOp);
58 REGISTER_XLA_OP(Name("ReluGrad"), MlirXlaOpKernel);
78 REGISTER_XLA_OP(Name("Relu6Grad"), Relu6GradOp);
95 REGISTER_XLA_OP(Name("LeakyReluGrad"), LeakyReluGradOp);
Dreduction_ops.cc44 REGISTER_XLA_OP(Name("Sum").CompileTimeConstantInput("reduction_indices"),
63 REGISTER_XLA_OP(Name("Prod").CompileTimeConstantInput("reduction_indices"),
81 REGISTER_XLA_OP(Name("Min").CompileTimeConstantInput("reduction_indices"),
113 REGISTER_XLA_OP(Name("Max").CompileTimeConstantInput("reduction_indices"),
148 REGISTER_XLA_OP(Name("Mean").CompileTimeConstantInput("reduction_indices"),
166 REGISTER_XLA_OP(Name("All").CompileTimeConstantInput("reduction_indices"),
184 REGISTER_XLA_OP(Name("Any").CompileTimeConstantInput("reduction_indices"),
Dxla_reduce_op.cc31 REGISTER_XLA_OP(Name("XlaReduce"), MlirXlaOpKernel);
32 REGISTER_XLA_OP(Name("XlaVariadicReduce"), MlirXlaOpKernel);
33 REGISTER_XLA_OP(Name("XlaVariadicReduceV2"), MlirXlaOpKernel);
Dshape_op.cc69 REGISTER_XLA_OP(Name("Shape").CompilationOnly().IsMetadataOp(), ShapeOp);
110 REGISTER_XLA_OP(Name("XlaSetBound").CompileTimeConstantInput("bound"),
141 REGISTER_XLA_OP(
169 REGISTER_XLA_OP(
211 REGISTER_XLA_OP(Name("ShapeN").CompilationOnly().IsMetadataOp(), ShapeNOp);
227 REGISTER_XLA_OP(Name("Rank").CompilationOnly().IsMetadataOp(), RankOp);
254 REGISTER_XLA_OP(Name("Size").CompilationOnly().IsMetadataOp(), SizeOp);
299 REGISTER_XLA_OP(Name("ExpandDims").CompileTimeConstantInput("dim"),
367 REGISTER_XLA_OP(Name("Squeeze"), SqueezeOp);
432 REGISTER_XLA_OP(Name("ZerosLike").AllowVariantTypes(), ZerosLikeOp);
[all …]
Dsort_ops.cc27 REGISTER_XLA_OP(Name("XlaSort"), MlirXlaOpKernel);
45 REGISTER_XLA_OP(Name("XlaKeyValueSort"), XlaKeyValueSortOp);
46 REGISTER_XLA_OP(Name("XlaVariadicSort").CompileTimeConstantInput("dimension"),
Dtensor_list_ops.cc107 REGISTER_XLA_OP(Name("TensorListLength").IsMetadataOp(), TensorListLengthOp);
208 REGISTER_XLA_OP(Name("TensorListReserve")
282 REGISTER_XLA_OP(Name("EmptyTensorList")
344 REGISTER_XLA_OP(Name("TensorListElementShape").IsMetadataOp(),
383 REGISTER_XLA_OP(Name("TensorListGetItem"), TensorListGetItemOp);
436 REGISTER_XLA_OP(Name("TensorListGather"), TensorListGatherOp);
466 REGISTER_XLA_OP(Name("TensorListStack"), TensorListStackOp);
521 REGISTER_XLA_OP(Name("TensorListConcatV2"), TensorListConcatOp);
577 REGISTER_XLA_OP(Name("TensorListSplit")
602 REGISTER_XLA_OP(
[all …]
Dbatch_matmul_op.cc57 REGISTER_XLA_OP(Name("BatchMatMul"), BatchMatMulOp);
58 REGISTER_XLA_OP(Name("BatchMatMulV2"), BatchMatMulOp);
59 REGISTER_XLA_OP(Name("BatchMatMulV3"), BatchMatMulOp);
Delu_op.cc74 REGISTER_XLA_OP(Name("Elu"), EluOp);
75 REGISTER_XLA_OP(Name("EluGrad"), EluGradOp);
107 REGISTER_XLA_OP(Name("Selu"), SeluOp);
108 REGISTER_XLA_OP(Name("SeluGrad"), SeluGradOp);
Dscatter_nd_op.cc142 REGISTER_XLA_OP(Name("ScatterNd").CompileTimeConstantInput("shape"),
253 REGISTER_XLA_OP(Name("TensorScatterAdd"), TensorScatterAddOp);
254 REGISTER_XLA_OP(Name("TensorScatterMax"), TensorScatterMaxOp);
255 REGISTER_XLA_OP(Name("TensorScatterMin"), TensorScatterMinOp);
256 REGISTER_XLA_OP(Name("TensorScatterSub"), TensorScatterSubOp);
257 REGISTER_XLA_OP(Name("TensorScatterUpdate"), TensorScatterUpdateOp);
Dbias_ops.cc76 REGISTER_XLA_OP(Name("BiasAdd"), BiasOp);
77 REGISTER_XLA_OP(Name("BiasAddV1"), BiasOp);
78 REGISTER_XLA_OP(Name("BiasAddGrad"), MlirXlaOpKernel);
Drandom_ops.cc75 REGISTER_XLA_OP(Name("RandomUniform").CompileTimeConstantInput("shape"),
78 REGISTER_XLA_OP(Name("RandomShuffle"), MlirXlaOpKernel);
114 REGISTER_XLA_OP(Name("RandomUniformInt").CompileTimeConstantInput("shape"),
147 REGISTER_XLA_OP(Name("RandomStandardNormal").CompileTimeConstantInput("shape"),
178 REGISTER_XLA_OP(Name("TruncatedNormal")
255 REGISTER_XLA_OP(Name("ParameterizedTruncatedNormal")
Dpooling_ops.cc240 REGISTER_XLA_OP(Name("MaxPool"), MaxPool2DOp);
241 REGISTER_XLA_OP(Name("MaxPoolV2")
251 REGISTER_XLA_OP(Name("MaxPool3D"), MaxPool3DOp);
301 REGISTER_XLA_OP(Name("AvgPool"), AvgPool2DOp);
303 REGISTER_XLA_OP(Name("AvgPool3D"), MlirXlaOpKernel);
431 REGISTER_XLA_OP(Name("MaxPoolGrad"), MaxPool2DGradOp);
432 REGISTER_XLA_OP(Name("MaxPoolGradV2")
437 REGISTER_XLA_OP(Name("MaxPool3DGrad"), MlirXlaOpKernel);
528 REGISTER_XLA_OP(
537 REGISTER_XLA_OP(
[all …]
Dmatrix_diag_ops.cc333 REGISTER_XLA_OP(Name("MatrixDiag"), MlirXlaOpKernel);
334 REGISTER_XLA_OP(Name("MatrixDiagV2")
340 REGISTER_XLA_OP(Name("MatrixDiagV3")
447 REGISTER_XLA_OP(Name("MatrixDiagPart"), MatrixDiagPartOp);
448 REGISTER_XLA_OP(Name("MatrixDiagPartV2")
452 REGISTER_XLA_OP(Name("MatrixDiagPartV3")
541 REGISTER_XLA_OP(Name("MatrixSetDiag"), MatrixSetDiagOp);
542 REGISTER_XLA_OP(Name("MatrixSetDiagV2").CompileTimeConstantInput("k"),
544 REGISTER_XLA_OP(Name("MatrixSetDiagV3").CompileTimeConstantInput("k"),
Dbatch_norm_op.cc237 REGISTER_XLA_OP(Name("FusedBatchNorm"), FusedBatchNormOp);
238 REGISTER_XLA_OP(Name("FusedBatchNormV2"), FusedBatchNormOp);
239 REGISTER_XLA_OP(Name("FusedBatchNormV3"), FusedBatchNormOpV3);
240 REGISTER_XLA_OP(Name("_FusedBatchNormEx"), FusedBatchNormOpEx);
354 REGISTER_XLA_OP(Name("FusedBatchNormGrad"), FusedBatchNormGradOp);
355 REGISTER_XLA_OP(Name("FusedBatchNormGradV2"), FusedBatchNormGradOp);
356 REGISTER_XLA_OP(Name("FusedBatchNormGradV3"), FusedBatchNormGradOp);
Dresampler_addon_ops.cc23 REGISTER_XLA_OP(Name("Addons>Resampler")
27 REGISTER_XLA_OP(Name("Addons>ResamplerGrad")
Dxla_conv_op.cc29 REGISTER_XLA_OP(Name("XlaConv")
37 REGISTER_XLA_OP(Name("XlaConvV2")
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry_test.cc56 REGISTER_XLA_OP(Name("DummyDuplicateOp")
62 REGISTER_XLA_OP(Name("DummyDuplicateOp").TypeConstraint("T", DT_FLOAT),
105 REGISTER_XLA_OP(
Dlight_outside_compilation_kernels_for_test.cc61 REGISTER_XLA_OP(Name("TestStaticTf").Device(DEVICE_GPU_XLA_JIT),
107 REGISTER_XLA_OP(Name("TestStaticMultipleOutputTf").Device(DEVICE_GPU_XLA_JIT),
174 REGISTER_XLA_OP(Name("TestDynamicTf").Device(DEVICE_GPU_XLA_JIT),
235 REGISTER_XLA_OP(Name("DynamicMultidim")
247 REGISTER_XLA_OP(Name("DynamicUnranked").Device(DEVICE_GPU_XLA_JIT),
301 REGISTER_XLA_OP(Name("TestTfMustBeConstant")
365 REGISTER_XLA_OP(Name("TestDynamicTfWithBound").Device(DEVICE_GPU_XLA_JIT),
/external/tensorflow/tensorflow/core/tpu/kernels/
Dimage_resize_ops.cc137 REGISTER_XLA_OP(Name("ResizeNearestNeighbor")
142 REGISTER_XLA_OP(Name("ResizeNearestNeighborGrad")
147 REGISTER_XLA_OP(Name("ResizeBilinear")
152 REGISTER_XLA_OP(Name("ResizeBilinearGrad").Device(DEVICE_TPU_XLA_JIT),

123456