op { | |
name: "BoostedTreesCenterBias" | |
input_arg { | |
name: "tree_ensemble_handle" | |
type: DT_RESOURCE | |
} | |
input_arg { | |
name: "mean_gradients" | |
type: DT_FLOAT | |
} | |
input_arg { | |
name: "mean_hessians" | |
type: DT_FLOAT | |
} | |
input_arg { | |
name: "l1" | |
type: DT_FLOAT | |
} | |
input_arg { | |
name: "l2" | |
type: DT_FLOAT | |
} | |
output_arg { | |
name: "continue_centering" | |
type: DT_BOOL | |
} | |
is_stateful: true | |
} |