load( "@xla//xla/tsl:tsl.bzl", "check_deps", ) load("@xla//xla/tsl/platform:build_config_root.bzl", "if_static") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cuda_library", ) load("//tensorflow:tensorflow.default.bzl", "tf_kernel_library") load( "//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library", ) load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], default_visibility = [ "//tensorflow/compiler/tf2xla:internal", "//third_party/cloud_tpu/inference_converter:__subpackages__", "//third_party/py/orbax:__subpackages__", ], licenses = ["notice"], ) check_deps( name = "xla_ops_check_deps", disallowed_deps = [ "@xla//xla/service/cpu:cpu_compiler", "@xla//xla/service/gpu:gpu_compiler", ], deps = [":xla_ops"], ) cc_library( name = "xla_ops", tags = ["optonly"], visibility = ["//visibility:public"], deps = [ ":aggregate_ops", ":all_reduce_op", ":approx_topk_op", ":arg_op", ":batch_matmul_op", ":batch_norm_op", ":batchtospace_op", ":bcast_ops", ":beta_op", ":bias_ops", ":binary_ops", ":bincount_op", ":broadcast_to_op", ":bucketize_op", ":case_op", ":cast_op", ":categorical_op", ":cholesky_op", ":clip_by_value_op", ":concat_op", ":const_op", ":conv_ops", ":cross_op", ":cwise_ops", ":data_format_ops", ":depthtospace_op", ":dequantize_op", ":device_index_op", ":diag_op", ":dynamic_partition_op", ":dynamic_slice_ops", ":dynamic_stitch_op", ":einsum_op", ":elu_op", ":empty_op", ":ensure_shape_op", ":extract_image_patches_op", ":fake_param_op", ":fake_quantize_ops", ":fft_ops", ":fill_op", ":function_ops", ":fused_conv_ops", ":gather_op", ":gather_scatter_ops", ":identity_op", ":if_op", ":image_ops", ":image_resize_ops", ":in_topk_op", ":index_ops", ":l2loss_op", ":listdiff_op", ":lower_upper_bound_ops", ":lrn_ops", ":matmul_op", ":matrix_band_part_op", ":matrix_diag_ops", ":matrix_inverse_op", ":matrix_solve_op", ":matrix_triangular_solve_op", ":mirror_pad_op", ":next_after_op", ":no_op", ":one_hot_op", ":pack_op", ":pad_op", ":pooling_ops", ":qr_op", ":quantize_and_dequantize_op", ":random_ops", ":random_ops_util", ":reduce_window_op", ":reduction_ops", ":reduction_ops_common", ":relu_op", ":replica_id_op", ":reshape_op", ":retval_op", ":reverse_op", ":reverse_sequence_op", ":roll_op", ":scan_ops", ":scatter_nd_op", ":segment_reduction_ops", ":select_op", ":sendrecv_ops", ":sequence_ops", ":shape_op", ":shape_util", ":sharding_op", ":sharding_util_ops", ":slice_op", ":softmax_op", ":sort_ops", ":spacetobatch_op", ":spacetodepth_op", ":sparse_to_dense_op", ":split_op", ":spmd_manual_sharding_ops", ":stack_ops", ":stateful_random_ops", ":stateless_random_ops", ":stateless_random_ops_v2", ":stochastic_cast_op", ":strided_slice_op", ":tensor_array_ops", ":tensor_list_ops", ":tile_ops", ":to_bool_op", ":topk_op", ":training_ops", ":transpose_op", ":tridiagonal_ops", ":unary_ops", ":unary_ops_composition", ":uniform_quantized_ops", ":unique_op", ":unpack_op", ":variable_ops", ":where_op", ":while_op", ":xla_broadcast_helper_op", ":xla_call_module_op", ":xla_conv_op", ":xla_custom_call_op", ":xla_custom_call_v2_op", ":xla_dequantize_op", ":xla_dot_op", ":xla_optimization_barrier_op", ":xla_pad_op", ":xla_reduce_op", ":xla_reduce_precision_op", ":xla_select_and_scatter_op", ":xla_self_adjoint_eig_op", ":xla_svd_op", ], ) tf_cuda_library( name = "light_outside_compilation", srcs = ["light_outside_compilation.cc"], hdrs = ["light_outside_compilation.h"], visibility = ["//visibility:public"], deps = [ ":callback_proto_cc", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/core:framework", "//tensorflow/core:gpu_runtime", "//tensorflow/core:protos_all_cc", "//tensorflow/core/common_runtime:process_state", "//tensorflow/core/platform:env", "//tensorflow/core/platform:refcount", "//tensorflow/core/platform:status", "//tensorflow/core/platform:types", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/base:core_headers", "@com_google_absl//absl/cleanup", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", "@com_google_absl//absl/memory", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/synchronization", "@com_google_absl//absl/types:span", "@tsl//tsl/platform:fingerprint", "@xla//xla:shape_util", "@xla//xla:status_macros", "@xla//xla:util", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/service:custom_call_status", "@xla//xla/service:custom_call_target_registry", "@xla//xla/service:hlo_proto_cc", "@xla//xla/stream_executor:device_memory", "@xla//xla/stream_executor:platform", "@xla//xla/stream_executor:platform_manager", "@xla//xla/stream_executor:stream", "@xla//xla/stream_executor:stream_executor_h", "@xla//xla/stream_executor:stream_finder", "@xla//xla/tsl/lib/strings:proto_serialization", "@xla//xla/tsl/platform:errors", "@xla//xla/tsl/platform:statusor", ], alwayslink = 1, ) # A separate cc_library for resampler_ops is needed because resampler is in # contrib/, and thus the declaration of resampler cannot be pulled into the deps # of xla_ops. Therefore, resampler_ops is its own cc_library target, and its # corresponding tf_kernel_library is defined in contrib/resampler/BUILD. cc_library( name = "resampler_ops", srcs = ["resampler_ops.cc"], hdrs = ["resampler_ops.h"], visibility = ["//visibility:public"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", ], alwayslink = 1, ) # See note on resampler_ops target for why this is also a seprate op. cc_library( name = "resampler_addon_ops", srcs = ["resampler_addon_ops.cc"], visibility = ["//visibility:public"], deps = [ ":resampler_ops", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/core:protos_all_cc", ], alwayslink = 1, ) cc_library( name = "conv_op_helpers", srcs = ["conv_op_helpers.cc"], hdrs = ["conv_op_helpers.h"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/framework:bounds_check", "//tensorflow/core/kernels:conv_grad_shape_utils", "//tensorflow/core/platform:statusor", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@tsl//tsl/platform:tensor_float_32_hdr_lib", "@xla//xla:literal_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", ] + if_static(["@tsl//tsl/platform:tensor_float_32_utils"]), ) cc_library( name = "tensor_list_utils", srcs = ["tensor_list_utils.cc"], hdrs = ["tensor_list_utils.h"], deps = [ "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_expression", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@xla//xla:literal_util", "@xla//xla:shape_util", "@xla//xla:status_macros", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) cc_library( name = "if_while_utils", srcs = ["if_while_utils.cc"], hdrs = ["if_while_utils.h"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_expression", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core/common_runtime:function_body", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@xla//xla:literal", "@xla//xla/hlo/builder:value_inference", ], ) cc_library( name = "xla_call_module_loader", srcs = ["xla_call_module_loader.cc"], hdrs = ["xla_call_module_loader.h"], deps = [ "//tensorflow/compiler/jit:flags", "//tensorflow/compiler/mlir/tensorflow:dump_mlir_util", "//tensorflow/compiler/mlir/tensorflow:error_util", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", "@com_google_absl//absl/log:vlog_is_on", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@llvm-project//llvm:Support", "@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:IR", "@llvm-project//mlir:Parser", "@llvm-project//mlir:Pass", "@llvm-project//mlir:Support", "@llvm-project//mlir:Transforms", "@shardy//shardy/dialect/sdy/ir:dialect", "@shardy//shardy/dialect/sdy/transforms/import:passes", "@stablehlo//:chlo_ops", "@stablehlo//:stablehlo_ops", "@stablehlo//:stablehlo_passes", "@stablehlo//:stablehlo_passes_optimization", "@stablehlo//:stablehlo_serialization", "@stablehlo//:vhlo_ops", "@xla//xla:shape_util", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/translate:stablehlo", "@xla//xla/mlir/utils:type_util", "@xla//xla/python:refine_polymorphic_shapes", "@xla//xla/service:hlo_proto_cc", "@xla//xla/service/spmd/shardy/sdy_round_trip:pipelines", "@xla//xla/tsl/platform:errors", "@xla//xla/tsl/platform:statusor", ], ) tf_kernel_library( name = "xla_call_module_op", srcs = ["xla_call_module_op.cc"], deps = [ ":xla_call_module_loader", "//tensorflow/compiler/mlir/tensorflow:convert_type", "//tensorflow/compiler/mlir/tensorflow:dump_mlir_util", "//tensorflow/compiler/tf2xla:side_effect_util", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:types", "//tensorflow/core/tpu:tpu_defs", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@llvm-project//llvm:Support", "@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:FuncExtensions", "@llvm-project//mlir:IR", "@llvm-project//mlir:Pass", "@llvm-project//mlir:Support", "@stablehlo//:stablehlo_ops", "@xla//xla:debug_options_flags", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/ir:hlo", "@xla//xla/hlo/translate:stablehlo", "@xla//xla/hlo/translate/mhlo_to_hlo:type_to_shape", "@xla//xla/service:hlo_module_config", "@xla//xla/service/spmd/shardy/stablehlo_round_trip:stablehlo_import", "@xla//xla/tsl/framework/mlir:status_scoped_diagnostic_handler", "@xla//xla/tsl/platform:errors", "@xla//xla/tsl/platform:statusor", ], ) tf_kernel_library( name = "while_op", srcs = ["while_op.cc"], hdrs = ["while_op.h"], deps = [ ":if_while_utils", ":tensor_list_utils", "//tensorflow/compiler/tf2xla:side_effect_util", "//tensorflow/compiler/tf2xla:tf2xla_util", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:status", "//tensorflow/core/platform:types", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:shape_tree", "@xla//xla:shape_util", "@xla//xla:status_macros", "@xla//xla:xla_data_proto_cc", "@xla//xla/client", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:tuple", "@xla//xla/tsl/platform:errors", "@xla//xla/tsl/platform:statusor", ], ) tf_kernel_library( name = "if_op", srcs = ["if_op.cc"], hdrs = ["if_op.h"], deps = [ ":if_while_utils", "//tensorflow/compiler/tf2xla:side_effect_util", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/common_runtime:function_body", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:status", "//tensorflow/core/platform:types", "@com_google_absl//absl/log", "@com_google_absl//absl/log:vlog_is_on", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:shape_util", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", "@xla//xla/tsl/platform:errors", ], ) tf_kernel_library( name = "case_op", srcs = ["case_op.cc"], hdrs = ["case_op.h"], deps = [ ":if_while_utils", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:side_effect_util", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log", "@com_google_absl//absl/log:vlog_is_on", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", ], ) # Kernels that have a dummy (no-op) implementation. tf_kernel_library( name = "xla_dummy_ops", srcs = [ "assert_op.cc", "check_numerics_op.cc", ], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/core:array_ops_op_lib", "//tensorflow/core:lib", "//tensorflow/core:logging_ops_op_lib", "@com_google_absl//absl/log", ], alwayslink = 1, ) tf_proto_library( name = "callback_proto", srcs = ["callback.proto"], protodeps = tf_additional_all_protos(), ) cc_library( name = "rng_converter_utils", srcs = ["rng_converter_utils.cc"], hdrs = ["rng_converter_utils.h"], deps = [ "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/core:framework", "@com_google_absl//absl/strings:string_view", "@xla//xla:xla_data_proto_cc", ], ) tf_kernel_library( name = "xla_dot_op", srcs = ["xla_dot_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "training_ops", srcs = ["training_ops.cc"], deps = [ ":cwise_ops", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "batch_matmul_op", srcs = ["batch_matmul_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:util", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:protos_all_cc", "@tsl//tsl/platform:tensor_float_32_hdr_lib", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder/lib:math", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "softmax_op", srcs = ["softmax_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "in_topk_op", srcs = ["in_topk_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:literal", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:sorting", ], ) tf_kernel_library( name = "unary_ops_composition", srcs = ["unary_ops_composition.cc"], deps = [ ":cwise_ops", ":elu_op", ":relu_op", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "topk_op", srcs = ["topk_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status:statusor", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:sorting", ], ) tf_kernel_library( name = "tensor_array_ops", srcs = ["tensor_array_ops.cc"], deps = [ ":gather_op", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:status_macros", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "tile_ops", srcs = ["tile_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/types:span", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "strided_slice_op", srcs = ["strided_slice_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", "@com_google_absl//absl/types:span", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", ], ) tf_kernel_library( name = "xla_broadcast_helper_op", srcs = ["xla_broadcast_helper_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "xla_svd_op", srcs = ["xla_svd_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:slicing", "@xla//xla/hlo/builder/lib:svd", ], ) tf_kernel_library( name = "reduction_ops", srcs = ["reduction_ops.cc"], hdrs = ["reduction_ops.h"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:status", "@com_google_absl//absl/status", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "batchtospace_op", srcs = ["batchtospace_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "spmd_manual_sharding_ops", srcs = ["spmd_manual_sharding_ops.cc"], deps = [ "//tensorflow/compiler/mlir/tensorflow:xla_sharding_util", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:types", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "fused_conv_ops", srcs = ["fused_conv_ops.cc"], deps = [ ":conv_op_helpers", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "matrix_band_part_op", srcs = ["matrix_band_part_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", ], ) tf_kernel_library( name = "clip_by_value_op", srcs = ["clip_by_value_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "sharding_util_ops", srcs = ["sharding_util_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "sort_ops", srcs = ["sort_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:comparators", ], ) tf_kernel_library( name = "matrix_inverse_op", srcs = ["matrix_inverse_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:matrix", "@xla//xla/hlo/builder/lib:qr", ], ) tf_kernel_library( name = "reduce_window_op", srcs = ["reduce_window_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", ], ) tf_kernel_library( name = "pad_op", srcs = ["pad_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@xla//xla:literal", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "function_ops", srcs = ["function_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", ], ) tf_kernel_library( name = "sparse_to_dense_op", srcs = ["sparse_to_dense_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:scatter", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "reverse_op", srcs = ["reverse_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/container:inlined_vector", "@xla//xla:literal", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "random_ops", srcs = ["random_ops.cc"], deps = [ ":gather_op", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/lib:random", "//tensorflow/compiler/tf2xla/lib:util", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@com_google_absl//absl/log", "@com_google_absl//absl/status:statusor", "@xla//xla:shape_util", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:comparators", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", "@xla//xla/hlo/builder/lib:loops", "@xla//xla/tsl/platform:statusor", ], ) tf_kernel_library( name = "gather_op", srcs = ["gather_op.cc"], hdrs = ["gather_op_helpers.h"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:optional", "@xla//xla:status_macros", "@xla//xla:xla_data_proto_cc", "@xla//xla/client:client_library", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:slicing", ], ) tf_kernel_library( name = "segment_reduction_ops", srcs = ["segment_reduction_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:scatter", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "dynamic_partition_op", srcs = ["dynamic_partition_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core/tpu:tpu_defs", "@com_google_absl//absl/algorithm:container", "@xla//xla:comparison_util", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:comparators", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "transpose_op", srcs = ["transpose_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:scatter", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:status", "//tensorflow/core/platform:types", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status", "@xla//xla:shape_util", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "identity_op", srcs = ["identity_op.cc"], deps = [ ":tensor_list_utils", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log:check", ], ) tf_kernel_library( name = "tensor_list_ops", srcs = ["tensor_list_ops.cc"], deps = [ ":gather_op", ":tensor_list_utils", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "cross_op", srcs = ["cross_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "next_after_op", srcs = ["next_after_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "stochastic_cast_op", srcs = ["stochastic_cast_op.cc"], deps = [ ":random_ops_util", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/kernels:stochastic_cast_op_header", "@com_google_absl//absl/status:statusor", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "select_op", srcs = ["select_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/algorithm:container", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "l2loss_op", srcs = ["l2loss_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "arg_op", srcs = ["arg_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:literal_util", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "shape_util", srcs = ["shape_util.cc"], hdrs = ["shape_util.h"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:status", "//tensorflow/core/platform:types", "@com_google_absl//absl/status", ], ) tf_kernel_library( name = "to_bool_op", srcs = ["to_bool_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/status", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "mirror_pad_op", srcs = ["mirror_pad_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status:statusor", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:status_macros", "@xla//xla:util", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "xla_reduce_op", srcs = ["xla_reduce_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", ], ) tf_kernel_library( name = "qr_op", srcs = ["qr_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:qr", ], ) tf_kernel_library( name = "slice_op", srcs = ["slice_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/types:span", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", ], ) tf_kernel_library( name = "pack_op", srcs = ["pack_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@xla//xla:literal_util", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "all_reduce_op", srcs = ["all_reduce_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "sharding_op", srcs = ["sharding_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:sharding_util", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@xla//xla:sharding_op_util", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "concat_op", srcs = ["concat_op.cc"], deps = [ ":shape_util", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:literal_util", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "scan_ops", srcs = ["scan_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", ], ) tf_kernel_library( name = "image_resize_ops", srcs = ["image_resize_ops.cc"], hdrs = ["image_resize_ops.h"], deps = [ "//tensorflow/compiler/jit:xla_activity_listener", "//tensorflow/compiler/jit:xla_activity_proto_cc", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/types:span", "@xla//xla:array4d", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "spacetobatch_op", srcs = ["spacetobatch_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/util:overflow", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/types:span", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "lrn_ops", srcs = ["lrn_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status", "@xla//xla/hlo/builder:padding", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "binary_ops", srcs = ["binary_ops.cc"], deps = [ ":cwise_ops", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/types:span", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/client:client_library", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "roll_op", srcs = ["roll_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/strings", "@xla//xla:literal", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:slicing", ], ) tf_kernel_library( name = "random_ops_util", srcs = ["random_ops_util.cc"], hdrs = ["random_ops_util.h"], deps = [ ":rng_converter_utils", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core/kernels:stateless_random_ops_v2_header", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:prng", ], ) tf_kernel_library( name = "unary_ops", srcs = ["unary_ops.cc"], deps = [ ":cwise_ops", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/status:statusor", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", "@xla//xla/tsl/platform:statusor", ], ) tf_kernel_library( name = "cwise_ops", srcs = ["cwise_ops.cc"], hdrs = ["cwise_ops.h"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:shape_util", "@xla//xla/client:client_library", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "matrix_triangular_solve_op", srcs = ["matrix_triangular_solve_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "relu_op", srcs = ["relu_op.cc"], hdrs = ["relu_op.h"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@xla//xla:literal", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "reduction_ops_common", srcs = ["reduction_ops_common.cc"], deps = [ ":reduction_ops", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", ], ) tf_kernel_library( name = "bucketize_op", srcs = ["bucketize_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", ], ) tf_kernel_library( name = "depthtospace_op", srcs = ["depthtospace_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:data_format", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "xla_optimization_barrier_op", srcs = ["xla_optimization_barrier_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", ], ) tf_kernel_library( name = "matmul_op", srcs = ["matmul_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@tsl//tsl/platform:tensor_float_32_hdr_lib", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "matrix_solve_op", srcs = ["matrix_solve_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:matrix", "@xla//xla/hlo/builder/lib:qr", ], ) tf_kernel_library( name = "categorical_op", srcs = ["categorical_op.cc"], deps = [ ":random_ops_util", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:prng", ], ) tf_kernel_library( name = "dynamic_stitch_op", srcs = ["dynamic_stitch_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:literal_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "beta_op", srcs = ["beta_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:status_macros", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:loops", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "unique_op", srcs = ["unique_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:optional", "@com_google_absl//absl/types:span", "@xla//xla:comparison_util", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:comparators", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "reshape_op", srcs = ["reshape_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/log", "@xla//xla:literal", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "pooling_ops", srcs = ["pooling_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:padding", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:pooling", ], ) tf_kernel_library( name = "data_format_ops", srcs = ["data_format_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:slicing", ], ) tf_kernel_library( name = "xla_dequantize_op", srcs = ["xla_dequantize_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:types", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:quantize", ], ) tf_kernel_library( name = "const_op", srcs = ["const_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "shape_op", srcs = ["shape_op.cc"], deps = [ ":shape_util", ":tensor_list_utils", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:literal", "@xla//xla:literal_util", "@xla//xla:shape_util", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "image_ops", srcs = ["image_ops.cc"], deps = [ ":gather_op", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:util", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:comparators", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", "@xla//xla/hlo/builder/lib:loops", "@xla//xla/hlo/builder/lib:sorting", ], ) tf_kernel_library( name = "retval_op", srcs = ["retval_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@xla//xla:status_macros", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "xla_custom_call_v2_op", srcs = ["xla_custom_call_v2_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/service:hlo_proto_cc", ], ) tf_kernel_library( name = "listdiff_op", srcs = ["listdiff_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "sendrecv_ops", srcs = ["sendrecv_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:types", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "conv_ops", srcs = ["conv_ops.cc"], deps = [ ":conv_op_helpers", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:literal_util", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "dequantize_op", srcs = ["dequantize_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "ensure_shape_op", srcs = ["ensure_shape_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:literal", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "where_op", srcs = ["where_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:shape_util", "@xla//xla:status_macros", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:comparators", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", ], ) tf_kernel_library( name = "stack_ops", srcs = ["stack_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:shape_util", "@xla//xla:status_macros", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "xla_reduce_precision_op", srcs = ["xla_reduce_precision_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", ], ) tf_kernel_library( name = "diag_op", srcs = ["diag_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:util", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:matrix", "@xla//xla/hlo/builder/lib:pooling", ], ) tf_kernel_library( name = "index_ops", srcs = ["index_ops.cc"], hdrs = ["index_ops.h"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", ], ) tf_kernel_library( name = "lower_upper_bound_ops", srcs = ["lower_upper_bound_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "@com_google_absl//absl/status", "@xla//xla:comparison_util", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "spacetodepth_op", srcs = ["spacetodepth_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:data_format", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@com_google_absl//absl/log:check", "@com_google_absl//absl/types:span", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "empty_op", srcs = ["empty_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "bincount_op", srcs = ["bincount_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:comparators", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "tridiagonal_ops", srcs = ["tridiagonal_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@xla//xla/hlo/builder/lib:slicing", "@xla//xla/hlo/builder/lib:tridiagonal", ], ) tf_kernel_library( name = "device_index_op", srcs = ["device_index_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/strings", "@xla//xla/client:client_library", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "bcast_ops", srcs = ["bcast_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:literal", "@xla//xla/hlo/builder:value_inference", ], ) tf_kernel_library( name = "aggregate_ops", srcs = ["aggregate_ops.cc"], deps = [ ":tensor_list_utils", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:lib", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "split_op", srcs = ["split_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@xla//xla:shape_util", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "replica_id_op", srcs = ["replica_id_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "bias_ops", srcs = ["bias_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "xla_select_and_scatter_op", srcs = ["xla_select_and_scatter_op.cc"], deps = [ ":while_op", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", ], ) tf_kernel_library( name = "stateless_random_ops_v2", srcs = ["stateless_random_ops_v2.cc"], deps = [ ":random_ops_util", ":rng_converter_utils", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:random", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/kernels:stateless_random_ops_v2_header", "//tensorflow/core/platform:status", "//tensorflow/core/platform:types", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:dynamic_shaped_ops", "@xla//xla/hlo/builder/lib:prng", ], ) tf_kernel_library( name = "approx_topk_op", srcs = ["approx_topk_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/tpu:tpu_defs", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/types:span", "@xla//xla:literal_util", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:approx_topk", ], ) tf_kernel_library( name = "stateful_random_ops", srcs = ["stateful_random_ops.cc"], deps = [ ":random_ops_util", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:random", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core/kernels:stateful_random_ops_header", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:prng", ], ) tf_kernel_library( name = "no_op", srcs = ["no_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", ], ) tf_kernel_library( name = "xla_conv_op", srcs = ["xla_conv_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "@xla//xla:xla_data_proto_cc", ], ) tf_kernel_library( name = "broadcast_to_op", srcs = ["broadcast_to_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:lib", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "sequence_ops", srcs = ["sequence_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@com_google_absl//absl/status:statusor", "@xla//xla:literal", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "variable_ops", srcs = ["variable_ops.cc"], deps = [ ":gather_op", ":shape_util", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:scatter", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/kernels:resource_variable_util", "//tensorflow/core/kernels:scatter_nd_util", "//tensorflow/core/platform:status", "//tensorflow/core/platform:types", "@com_google_absl//absl/status", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:slicing", "@xla//xla/tsl/platform:errors", ], ) tf_kernel_library( name = "matrix_diag_ops", srcs = ["matrix_diag_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@eigen_archive//:eigen3", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "reverse_sequence_op", srcs = ["reverse_sequence_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "xla_custom_call_op", srcs = ["xla_custom_call_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/log:check", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "xla_self_adjoint_eig_op", srcs = ["xla_self_adjoint_eig_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@xla//xla/hlo/builder/lib:self_adjoint_eig", ], ) tf_kernel_library( name = "cast_op", srcs = ["cast_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/lib:util", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "dynamic_slice_ops", srcs = ["dynamic_slice_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "fft_ops", srcs = ["fft_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:literal_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "xla_pad_op", srcs = ["xla_pad_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "one_hot_op", srcs = ["one_hot_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:types", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "unpack_op", srcs = ["unpack_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "elu_op", srcs = ["elu_op.cc"], hdrs = ["elu_op.h"], deps = [ ":cwise_ops", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@xla//xla:literal", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", ], ) tf_kernel_library( name = "batch_norm_op", srcs = ["batch_norm_op.cc"], deps = [ ":relu_op", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "extract_image_patches_op", srcs = ["extract_image_patches_op.cc"], deps = [ ":conv_op_helpers", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "scatter_nd_op", srcs = ["scatter_nd_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:scatter", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:errors", "//tensorflow/core/platform:status", "@com_google_absl//absl/status", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "fill_op", srcs = ["fill_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:value_inference", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "stateless_random_ops", srcs = ["stateless_random_ops.cc"], deps = [ ":random_ops_util", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/lib:broadcast", "//tensorflow/compiler/tf2xla/lib:random", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:prng", ], ) tf_kernel_library( name = "gather_scatter_ops", srcs = ["gather_scatter_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", ], ) tf_kernel_library( name = "einsum_op", srcs = ["einsum_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "cholesky_op", srcs = ["cholesky_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:matrix", ], ) tf_kernel_library( name = "quantize_and_dequantize_op", srcs = ["quantize_and_dequantize_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/types:span", "@xla//xla:shape_util", "@xla//xla:xla_data_proto_cc", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder:xla_computation", "@xla//xla/hlo/builder/lib:arithmetic", "@xla//xla/hlo/builder/lib:constants", "@xla//xla/hlo/builder/lib:math", ], ) tf_kernel_library( name = "fake_quantize_ops", srcs = ["fake_quantize_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla/hlo/builder:xla_builder", "@xla//xla/hlo/builder/lib:arithmetic", ], ) tf_kernel_library( name = "fake_param_op", srcs = ["fake_param_op.cc"], deps = [ "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compilation_device", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", "//tensorflow/compiler/tf2xla:xla_helpers", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla:xla_resource", "//tensorflow/compiler/tf2xla/ops:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "@xla//xla/hlo/builder/lib:constants", ], ) tf_cc_test( name = "rng_converter_utils_test", srcs = ["rng_converter_utils_test.cc"], deps = [ ":rng_converter_utils", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/core:framework", "@com_google_googletest//:gtest_main", "@xla//xla:xla_data_proto_cc", ], ) tf_kernel_library( name = "uniform_quantized_ops", srcs = ["uniform_quantized_ops.cc"], deps = [ "//tensorflow/compiler/tf2xla:mlir_xla_op_kernel", "//tensorflow/compiler/tf2xla:xla_op_registry", ], )