blob: 49a82d2ba0c772e55255c031faa06559f45565de [file] [log] [blame]
op {
name: "BoostedTreesMakeStatsSummary"
input_arg {
name: "node_ids"
type: DT_INT32
}
input_arg {
name: "gradients"
type: DT_FLOAT
}
input_arg {
name: "hessians"
type: DT_FLOAT
}
input_arg {
name: "bucketized_features_list"
type: DT_INT32
number_attr: "num_features"
}
output_arg {
name: "stats_summary"
type: DT_FLOAT
}
attr {
name: "max_splits"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "num_buckets"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "num_features"
type: "int"
has_minimum: true
minimum: 1
}
}