Searched refs:REGISTER_XLA_OP (Results 1 – 25 of 136) sorted by relevance
123456
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 116 REGISTER_XLA_OP(Name("FFT").TypeConstraint("Tcomplex", DT_COMPLEX64), FFTOp<1>);
117 REGISTER_XLA_OP(Name("FFT2D").TypeConstraint("Tcomplex", DT_COMPLEX64),
119 REGISTER_XLA_OP(Name("FFT3D").TypeConstraint("Tcomplex", DT_COMPLEX64),
128 REGISTER_XLA_OP(Name("IFFT").TypeConstraint("Tcomplex", DT_COMPLEX64),
130 REGISTER_XLA_OP(Name("IFFT2D").TypeConstraint("Tcomplex", DT_COMPLEX64),
132 REGISTER_XLA_OP(Name("IFFT3D").TypeConstraint("Tcomplex", DT_COMPLEX64),
141 REGISTER_XLA_OP(Name("RFFT")
146 REGISTER_XLA_OP(Name("RFFT2D")
151 REGISTER_XLA_OP(Name("RFFT3D")
163 REGISTER_XLA_OP(Name("IRFFT")
[all …]
|
| D | variable_ops.cc | 41 REGISTER_XLA_OP(Name("VarIsInitializedOp"), VarIsInitializedOp);
62 REGISTER_XLA_OP(Name("VariableShape").CompilationOnly(), VariableShapeOp);
80 REGISTER_XLA_OP(Name("ReadVariableOp").CompilationOnly(), ReadVariableOp);
90 REGISTER_XLA_OP(Name("AssignVariableOp").CompilationOnly(), AssignVariableOp);
104 REGISTER_XLA_OP(
120 REGISTER_XLA_OP(
145 REGISTER_XLA_OP(Name("ResourceGather"), ResourceGatherOp);
195 REGISTER_XLA_OP(Name("ResourceScatterAdd"), ResourceScatterAddOp);
208 REGISTER_XLA_OP(Name("ResourceScatterSub"), ResourceScatterSubOp);
221 REGISTER_XLA_OP(Name("ResourceScatterMul"), ResourceScatterMulOp);
[all …]
|
| D | identity_op.cc | 47 REGISTER_XLA_OP(
50 REGISTER_XLA_OP(Name("IdentityN")
55 REGISTER_XLA_OP(Name("PlaceholderWithDefault"), IdentityOp);
56 REGISTER_XLA_OP(Name("PreventGradient"), IdentityOp);
57 REGISTER_XLA_OP(Name("StopGradient").AllowVariantTypes(), IdentityOp);
58 REGISTER_XLA_OP(Name("Snapshot"), IdentityOp);
|
| D | conv_ops.cc | 73 REGISTER_XLA_OP(Name("Conv2D").TypeConstraint("T", GetXlaConvTypes()),
81 REGISTER_XLA_OP(Name("Conv3D").TypeConstraint("T", GetXlaConvTypes()),
89 REGISTER_XLA_OP(
135 REGISTER_XLA_OP(Name("Conv2DBackpropInput")
145 REGISTER_XLA_OP(Name("Conv3DBackpropInputV2")
155 REGISTER_XLA_OP(Name("DepthwiseConv2dNativeBackpropInput")
197 REGISTER_XLA_OP(Name("Conv2DBackpropFilter")
208 REGISTER_XLA_OP(Name("Conv3DBackpropFilterV2")
218 REGISTER_XLA_OP(Name("DepthwiseConv2dNativeBackpropFilter")
|
| D | function_ops.cc | 51 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")
|
| D | relu_op.cc | 46 REGISTER_XLA_OP(Name("Relu"), ReluOp);
56 REGISTER_XLA_OP(Name("Relu6"), Relu6Op);
72 REGISTER_XLA_OP(Name("LeakyRelu"), LeakyReluOp);
88 REGISTER_XLA_OP(Name("ReluGrad"), ReluGradOp);
108 REGISTER_XLA_OP(Name("Relu6Grad"), Relu6GradOp);
125 REGISTER_XLA_OP(Name("LeakyReluGrad"), LeakyReluGradOp);
|
| D | reduction_ops.cc | 44 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"),
|
| D | shape_op.cc | 69 REGISTER_XLA_OP(Name("Shape").CompilationOnly().IsMetadataOp(), ShapeOp);
110 REGISTER_XLA_OP(Name("XlaSetBound").CompileTimeConstantInput("bound"),
141 REGISTER_XLA_OP(
183 REGISTER_XLA_OP(Name("ShapeN").CompilationOnly().IsMetadataOp(), ShapeNOp);
199 REGISTER_XLA_OP(Name("Rank").CompilationOnly().IsMetadataOp(), RankOp);
226 REGISTER_XLA_OP(Name("Size").CompilationOnly().IsMetadataOp(), SizeOp);
271 REGISTER_XLA_OP(Name("ExpandDims").CompileTimeConstantInput("dim"),
333 REGISTER_XLA_OP(Name("Squeeze"), SqueezeOp);
398 REGISTER_XLA_OP(Name("ZerosLike").AllowVariantTypes(), ZerosLikeOp);
412 REGISTER_XLA_OP(Name("OnesLike"), OnesLikeOp);
|
| D | tensor_list_ops.cc | 105 REGISTER_XLA_OP(Name("TensorListLength").IsMetadataOp(), TensorListLengthOp);
204 REGISTER_XLA_OP(Name("TensorListReserve")
276 REGISTER_XLA_OP(Name("EmptyTensorList")
336 REGISTER_XLA_OP(Name("TensorListElementShape").IsMetadataOp(),
375 REGISTER_XLA_OP(Name("TensorListGetItem"), TensorListGetItemOp);
428 REGISTER_XLA_OP(Name("TensorListGather"), TensorListGatherOp);
458 REGISTER_XLA_OP(Name("TensorListStack"), TensorListStackOp);
513 REGISTER_XLA_OP(Name("TensorListConcatV2"), TensorListConcatOp);
569 REGISTER_XLA_OP(Name("TensorListSplit")
594 REGISTER_XLA_OP(
[all …]
|
| D | scatter_nd_op.cc | 135 REGISTER_XLA_OP(Name("ScatterNd").CompileTimeConstantInput("shape"),
235 REGISTER_XLA_OP(Name("TensorScatterAdd"), TensorScatterAddOp);
236 REGISTER_XLA_OP(Name("TensorScatterMax"), TensorScatterMaxOp);
237 REGISTER_XLA_OP(Name("TensorScatterMin"), TensorScatterMinOp);
238 REGISTER_XLA_OP(Name("TensorScatterSub"), TensorScatterSubOp);
239 REGISTER_XLA_OP(Name("TensorScatterUpdate"), TensorScatterUpdateOp);
|
| D | elu_op.cc | 74 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);
|
| D | pooling_ops.cc | 197 REGISTER_XLA_OP(Name("MaxPool"), MaxPool2DOp);
198 REGISTER_XLA_OP(Name("MaxPoolV2")
208 REGISTER_XLA_OP(Name("MaxPool3D"), MaxPool3DOp);
258 REGISTER_XLA_OP(Name("AvgPool"), AvgPool2DOp);
265 REGISTER_XLA_OP(Name("AvgPool3D"), AvgPool3DOp);
383 REGISTER_XLA_OP(Name("MaxPoolGrad"), MaxPool2DGradOp);
384 REGISTER_XLA_OP(Name("MaxPoolGradV2")
394 REGISTER_XLA_OP(Name("MaxPool3DGrad"), MaxPool3DGradOp);
480 REGISTER_XLA_OP(
489 REGISTER_XLA_OP(
[all …]
|
| D | matrix_diag_ops.cc | 330 REGISTER_XLA_OP(Name("MatrixDiag"), MatrixDiagOp);
331 REGISTER_XLA_OP(Name("MatrixDiagV2")
337 REGISTER_XLA_OP(Name("MatrixDiagV3")
444 REGISTER_XLA_OP(Name("MatrixDiagPart"), MatrixDiagPartOp);
445 REGISTER_XLA_OP(Name("MatrixDiagPartV2")
449 REGISTER_XLA_OP(Name("MatrixDiagPartV3")
538 REGISTER_XLA_OP(Name("MatrixSetDiag"), MatrixSetDiagOp);
539 REGISTER_XLA_OP(Name("MatrixSetDiagV2").CompileTimeConstantInput("k"),
541 REGISTER_XLA_OP(Name("MatrixSetDiagV3").CompileTimeConstantInput("k"),
|
| D | batch_norm_op.cc | 233 REGISTER_XLA_OP(Name("FusedBatchNorm"), FusedBatchNormOp);
234 REGISTER_XLA_OP(Name("FusedBatchNormV2"), FusedBatchNormOp);
235 REGISTER_XLA_OP(Name("FusedBatchNormV3"), FusedBatchNormOpV3);
236 REGISTER_XLA_OP(Name("_FusedBatchNormEx"), FusedBatchNormOpEx);
350 REGISTER_XLA_OP(Name("FusedBatchNormGrad"), FusedBatchNormGradOp);
351 REGISTER_XLA_OP(Name("FusedBatchNormGradV2"), FusedBatchNormGradOp);
352 REGISTER_XLA_OP(Name("FusedBatchNormGradV3"), FusedBatchNormGradOp);
|
| D | no_op.cc | 34 REGISTER_XLA_OP(Name("NoOp").CompilationOnly(), NoOp);
38 REGISTER_XLA_OP(Name("ControlTrigger").CompilationOnly(), NoOp);
|
| D | random_ops.cc | 67 REGISTER_XLA_OP(Name("RandomUniform").CompileTimeConstantInput("shape"),
217 REGISTER_XLA_OP(Name("RandomShuffle"), RandomShuffleOp);
252 REGISTER_XLA_OP(Name("RandomUniformInt").CompileTimeConstantInput("shape"),
281 REGISTER_XLA_OP(Name("RandomStandardNormal").CompileTimeConstantInput("shape"),
311 REGISTER_XLA_OP(Name("TruncatedNormal")
358 REGISTER_XLA_OP(Name("ParameterizedTruncatedNormal")
|
| D | tensor_array_ops.cc | 188 REGISTER_XLA_OP(Name("TensorArrayV3").CompileTimeConstantInput("size"),
243 REGISTER_XLA_OP(Name("TensorArrayWriteV3"), TensorArrayWriteOp);
286 REGISTER_XLA_OP(Name("TensorArrayReadV3"), TensorArrayReadOp);
353 REGISTER_XLA_OP(Name("TensorArrayGatherV3"), TensorArrayGatherOp);
440 REGISTER_XLA_OP(Name("TensorArrayScatterV3"), TensorArrayScatterOp);
480 REGISTER_XLA_OP(Name("TensorArrayConcatV3"), TensorArrayConcatOp);
550 REGISTER_XLA_OP(Name("TensorArraySplitV3").CompileTimeConstantInput("lengths"),
569 REGISTER_XLA_OP(Name("TensorArraySizeV3"), TensorArraySizeOp);
604 REGISTER_XLA_OP(Name("TensorArrayGradV3"), TensorArrayGradOp);
618 REGISTER_XLA_OP(Name("TensorArrayCloseV3"), TensorArrayCloseOp);
|
| D | segment_reduction_ops.cc | 139 REGISTER_XLA_OP(
154 REGISTER_XLA_OP(
171 REGISTER_XLA_OP(
188 REGISTER_XLA_OP(
|
| D | batch_matmul_op.cc | 43 REGISTER_XLA_OP(Name("BatchMatMul"), BatchMatMulOp);
44 REGISTER_XLA_OP(Name("BatchMatMulV2"), BatchMatMulOp);
|
| D | bias_ops.cc | 75 REGISTER_XLA_OP(Name("BiasAdd"), BiasOp);
76 REGISTER_XLA_OP(Name("BiasAddV1"), BiasOp);
125 REGISTER_XLA_OP(Name("BiasAddGrad"), BiasAddGradOp);
|
| D | stack_ops.cc | 123 REGISTER_XLA_OP(
172 REGISTER_XLA_OP(Name("StackPushV2").CompilationOnly(), StackPushOp);
226 REGISTER_XLA_OP(Name("StackPopV2").CompilationOnly(), StackPopOp);
240 REGISTER_XLA_OP(Name("StackCloseV2").CompilationOnly(), StackCloseOp);
|
| D | training_ops.cc | 55 REGISTER_XLA_OP(Name("ResourceApplyGradientDescent")
113 REGISTER_XLA_OP(Name("ResourceApplyProximalGradientDescent")
171 REGISTER_XLA_OP(Name("ResourceApplyMomentum").TypeConstraint("T", kFloatTypes),
229 REGISTER_XLA_OP(Name("ResourceApplyKerasMomentum")
277 REGISTER_XLA_OP(
332 REGISTER_XLA_OP(
388 REGISTER_XLA_OP(Name("ResourceApplyProximalAdagrad")
465 REGISTER_XLA_OP(Name("ResourceApplyAdagradDA").TypeConstraint("T", kFloatTypes),
561 REGISTER_XLA_OP(
633 REGISTER_XLA_OP(Name("ResourceApplyAdaMax").TypeConstraint("T", kFloatTypes),
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry_test.cc | 56 REGISTER_XLA_OP(Name("DummyDuplicateOp")
62 REGISTER_XLA_OP(Name("DummyDuplicateOp").TypeConstraint("T", DT_FLOAT),
105 REGISTER_XLA_OP(
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 137 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),
|
| D | cross_replica_ops.cc | 127 REGISTER_XLA_OP(Name("AllToAll").CompileTimeConstantInput("group_assignment"),
129 REGISTER_XLA_OP(Name("CollectivePermute")
134 REGISTER_XLA_OP(
|
123456