Searched defs:feature_ids (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 72 def deserialize(self, stamp_token, num_updates, partition_ids, feature_ids, argument
196 def add(self, stamp_token, partition_ids, feature_ids, gradients, hessians): argument
209 def schedule_add(self, partition_ids, feature_ids, gradients, hessians): argument
230 def _make_summary(self, partition_ids, feature_ids, gradients, hessians): argument
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 82 const auto feature_ids = feature_ids_t->vec<int32>(); in Compute() local
199 const TTypes<const int32>::Vec& feature_ids, in FindBestSplitsPerNode()
|
| D | boosted_trees.proto | 166 repeated int32 feature_ids = 1; field
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/proto/ |
| D | tree_config.proto | 101 repeated int64 feature_ids = 2; field
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | split_handler_ops.cc | 718 const auto& feature_ids = feature_ids_t->matrix<int64>(); in Compute() local
817 const tensorflow::TTypes<int64>::ConstMatrix& feature_ids, in ComputeNormalDecisionTree()
890 const tensorflow::TTypes<int64>::ConstMatrix& feature_ids, in ComputeObliviousDecisionTree()
|
| D | stats_accumulator_ops.cc | 147 auto feature_ids = feature_ids_t->matrix<int64>(); in SerializeScalarAccumulatorToOutput() local
187 auto feature_ids = feature_ids_t->matrix<int64>(); in SerializeTensorAccumulatorToOutput() local
|