Fix ops namespace to boosted_trees.
PiperOrigin-RevId: 169251963
This commit is contained in:
parent
34018f8fa7
commit
095f6aa7d1
@ -19,7 +19,7 @@
|
||||
#include "tensorflow/core/framework/shape_inference.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace gtflow {
|
||||
namespace boosted_trees {
|
||||
|
||||
REGISTER_RESOURCE_HANDLE_OP(DecisionTreeEnsembleResource);
|
||||
|
||||
@ -110,5 +110,5 @@ stamp_token: Token to use as the new value of the resource stamp.
|
||||
tree_ensemble_config: Serialized proto of the ensemble.
|
||||
)doc");
|
||||
|
||||
} // namespace gtflow
|
||||
} // namespace boosted_trees
|
||||
} // namespace tensorflow
|
||||
|
@ -18,7 +18,7 @@
|
||||
#include "tensorflow/core/framework/shape_inference.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace gtflow {
|
||||
namespace boosted_trees {
|
||||
using shape_inference::InferenceContext;
|
||||
using shape_inference::DimensionHandle;
|
||||
using shape_inference::ShapeHandle;
|
||||
@ -314,5 +314,5 @@ of the buckets.
|
||||
output: Same shape as 'input', where each value of input is replaced with its corresponding bucket index.
|
||||
)doc");
|
||||
|
||||
} // namespace gtflow
|
||||
} // namespace boosted_trees
|
||||
} // namespace tensorflow
|
||||
|
@ -18,7 +18,7 @@
|
||||
#include "tensorflow/core/framework/shape_inference.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace gtflow {
|
||||
namespace boosted_trees {
|
||||
using shape_inference::InferenceContext;
|
||||
using shape_inference::ShapeHandle;
|
||||
using shape_inference::DimensionHandle;
|
||||
@ -471,5 +471,5 @@ output_gradients: A tensor of gradients, first dimension matches slots
|
||||
output_hessians: A tensor of hessians, first dimension matches slots
|
||||
in <partition_id, feature_id>.
|
||||
)doc");
|
||||
} // namespace gtflow
|
||||
} // namespace boosted_trees
|
||||
} // namespace tensorflow
|
||||
|
@ -17,7 +17,7 @@
|
||||
#include "tensorflow/core/framework/shape_inference.h"
|
||||
|
||||
namespace tensorflow {
|
||||
namespace gtflow {
|
||||
namespace boosted_trees {
|
||||
|
||||
REGISTER_OP("CenterTreeEnsembleBias")
|
||||
.Attr("learner_config: string")
|
||||
@ -116,5 +116,5 @@ active_tree: Scalar indicating the active tree being trained.
|
||||
active_layer: Scalar indicating the active layer being trained.
|
||||
)doc");
|
||||
|
||||
} // namespace gtflow
|
||||
} // namespace boosted_trees
|
||||
} // namespace tensorflow
|
||||
|
Loading…
Reference in New Issue
Block a user