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

Add Buckettize, SparseCross and BoostedTreesBucketize to flex whitelist

Browse Source 
PiperOrigin-RevId: 313690852
Change-Id: I9dc21e2439d103ea3783652fb38826e0fd76aaaf
This commit is contained in:
Thai Nguyen 2020-05-28 17:25:45 -07:00 committed by TensorFlower Gardener
parent 5c9dd54540
commit 12bda81b3d
5 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/BUILD
View File

@ -6617,6 +6617,7 @@ filegroup(
"batch_matmul_op_impl.h",
"batch_norm_op.h",
"broadcast_to_op.h",
"bucketize_op.h",
"control_flow_ops.h",
"conv_2d.h",
"conv_3d.h",
@ -6681,6 +6682,8 @@ filegroup(
"transpose_op.h",
"where_op.h",
"xent_op.h",
] + [
"//tensorflow/core/kernels/boosted_trees/quantiles:weighted_quantiles_hdrs",
],
)
@ -6810,6 +6813,7 @@ filegroup(
srcs = [
"batchtospace_op.cc",
"broadcast_to_op.cc",
"bucketize_op.cc",
"ctc_decoder_ops.cc",
"decode_bmp_op.cc",
"depthtospace_op.cc",
@ -6886,6 +6890,7 @@ filegroup(
"spacetobatch_functor.cc",
"spacetobatch_op.cc",
"spacetodepth_op.cc",
"sparse_cross_op.cc",
"sparse_fill_empty_rows_op.cc",
"sparse_reshape_op.cc",
"sparse_to_dense_op.cc",
@ -6934,6 +6939,8 @@ filegroup(
"where_op.cc",
"xent_op.cc",
":android_extended_ops_headers",
] + [
"//tensorflow/core/kernels/boosted_trees:quantile_ops.cc",
],
)
@ -7096,6 +7103,7 @@ cc_library(
deps = [
"//tensorflow/core:portable_tensorflow_lib_lite",
"//tensorflow/core:protos_all_cc_impl",
"//tensorflow/core/platform:strong_hash",
"//third_party/eigen3",
"//third_party/fft2d:fft2d_headers",
"@com_google_absl//absl/base",

2
tensorflow/core/kernels/boosted_trees/BUILD
View File

@ -19,6 +19,8 @@ package(
licenses = ["notice"], # Apache 2.0
)
exports_files(["quantile_ops.cc"])
tf_proto_library(
name = "boosted_trees_proto",
srcs = [

10
tensorflow/core/kernels/boosted_trees/quantiles/BUILD
View File

@ -10,6 +10,16 @@ package(
exports_files(["LICENSE"])
filegroup(
name = "weighted_quantiles_hdrs",
srcs = [
"quantile_stream_resource.h",
"weighted_quantiles_buffer.h",
"weighted_quantiles_stream.h",
"weighted_quantiles_summary.h",
],
)
# Quantiles
cc_library(

1
tensorflow/core/platform/BUILD
View File

@ -675,6 +675,7 @@ cc_library(
cc_library(
name = "strong_hash",
hdrs = ["strong_hash.h"],
visibility = ["//visibility:public"],
deps = [
":platform",
":types",

5
tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
View File

@ -66,6 +66,8 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
"BiasAdd",
"BiasAddGrad",
"BiasAddV1",
"BoostedTreesBucketize",
"Bucketize",
"BroadcastArgs",
"BroadcastGradientArgs",
"BroadcastTo",
@ -386,6 +388,9 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
"SparseApplyProximalAdagrad",
"SparseApplyProximalGradientDescent",
"SparseApplyRMSProp",
"SparseCross",
"SparseCrossHashed",
"SparseCrossV2",
"SparseFillEmptyRows",
"SparseFillEmptyRowsGrad",
"SparseReshape",