Fix ops namespace to boosted_trees.

PiperOrigin-RevId: 169251963
This commit is contained in:
A. Unique TensorFlower 2017-09-19 09:27:31 -07:00 committed by TensorFlower Gardener
parent 34018f8fa7
commit 095f6aa7d1
4 changed files with 8 additions and 8 deletions

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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