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

Update ops-related pbtxt files.

Browse Source 
PiperOrigin-RevId: 254122505
This commit is contained in:
A. Unique TensorFlower 2019-06-19 20:07:13 -07:00 committed by TensorFlower Gardener
parent ba0ecbce0a
commit 8a1ddd555e
2 changed files with 162 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
tensorflow/core/ops/compat/ops_history.v1.pbtxt
View File

@ -12518,6 +12518,87 @@ op {
minimum: 1
}
}
op {
name: "BoostedTreesSparseCalculateBestFeatureSplit"
input_arg {
name: "node_id_range"
type: DT_INT32
}
input_arg {
name: "stats_summary_indices"
type: DT_INT32
}
input_arg {
name: "stats_summary_values"
type: DT_FLOAT
}
input_arg {
name: "stats_summary_shape"
type: DT_INT32
}
input_arg {
name: "l1"
type: DT_FLOAT
}
input_arg {
name: "l2"
type: DT_FLOAT
}
input_arg {
name: "tree_complexity"
type: DT_FLOAT
}
input_arg {
name: "min_node_weight"
type: DT_FLOAT
}
output_arg {
name: "node_ids"
type: DT_INT32
}
output_arg {
name: "gains"
type: DT_FLOAT
}
output_arg {
name: "feature_dimensions"
type: DT_INT32
}
output_arg {
name: "thresholds"
type: DT_INT32
}
output_arg {
name: "left_node_contribs"
type: DT_FLOAT
}
output_arg {
name: "right_node_contribs"
type: DT_FLOAT
}
output_arg {
name: "split_with_default_directions"
type: DT_STRING
}
attr {
name: "logits_dimension"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "split_type"
type: "string"
default_value {
s: "inequality"
}
allowed_values {
list {
s: "inequality"
}
}
}
}
op {
name: "BoostedTreesTrainingPredict"
input_arg {

81
tensorflow/core/ops/ops.pbtxt
View File

@ -5273,6 +5273,87 @@ op {
minimum: 1
}
}
op {
name: "BoostedTreesSparseCalculateBestFeatureSplit"
input_arg {
name: "node_id_range"
type: DT_INT32
}
input_arg {
name: "stats_summary_indices"
type: DT_INT32
}
input_arg {
name: "stats_summary_values"
type: DT_FLOAT
}
input_arg {
name: "stats_summary_shape"
type: DT_INT32
}
input_arg {
name: "l1"
type: DT_FLOAT
}
input_arg {
name: "l2"
type: DT_FLOAT
}
input_arg {
name: "tree_complexity"
type: DT_FLOAT
}
input_arg {
name: "min_node_weight"
type: DT_FLOAT
}
output_arg {
name: "node_ids"
type: DT_INT32
}
output_arg {
name: "gains"
type: DT_FLOAT
}
output_arg {
name: "feature_dimensions"
type: DT_INT32
}
output_arg {
name: "thresholds"
type: DT_INT32
}
output_arg {
name: "left_node_contribs"
type: DT_FLOAT
}
output_arg {
name: "right_node_contribs"
type: DT_FLOAT
}
output_arg {
name: "split_with_default_directions"
type: DT_STRING
}
attr {
name: "logits_dimension"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "split_type"
type: "string"
default_value {
s: "inequality"
}
allowed_values {
list {
s: "inequality"
}
}
}
}
op {
name: "BoostedTreesTrainingPredict"
input_arg {