experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Define Tf_COMPILE_LIBRARY for two libraries (#15749)

Browse Source
This commit is contained in:
Loo Rong Jie 2018-01-01 08:06:52 +08:00 committed by drpngx
parent af367f55ed
commit a060e062cf
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/tf2xla/kernels/BUILD
View File

@ -4,6 +4,7 @@ package(
default_visibility = ["//tensorflow/compiler/tf2xla:internal"], default_visibility = ["//tensorflow/compiler/tf2xla:internal"],
) )
load("//tensorflow:tensorflow.bzl", "tf_copts")
load("//tensorflow:tensorflow.bzl", "tf_kernel_library") load("//tensorflow:tensorflow.bzl", "tf_kernel_library")
tf_kernel_library( tf_kernel_library(
@ -165,6 +166,7 @@ tf_kernel_library(
cc_library( cc_library(
name = "index_ops_kernel_argmax_float_1d", name = "index_ops_kernel_argmax_float_1d",
srcs = ["index_ops_kernel_argmax_float_1d.cc"], srcs = ["index_ops_kernel_argmax_float_1d.cc"],
copts = tf_copts(),
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/compiler/xla/service/cpu:custom_call_target_registry", "//tensorflow/compiler/xla/service/cpu:custom_call_target_registry",
@ -177,6 +179,7 @@ cc_library(
cc_library( cc_library(
name = "index_ops_kernel_argmax_float_2d", name = "index_ops_kernel_argmax_float_2d",
srcs = ["index_ops_kernel_argmax_float_2d.cc"], srcs = ["index_ops_kernel_argmax_float_2d.cc"],
copts = tf_copts(),
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//tensorflow/compiler/xla/service/cpu:custom_call_target_registry", "//tensorflow/compiler/xla/service/cpu:custom_call_target_registry",