Searched defs:partition_ids (Results 1 – 5 of 5) 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/contrib/boosted_trees/kernels/ |
| D | split_handler_ops.cc | 152 const auto& partition_ids = partition_ids_t->vec<int32>(); in Compute() local
246 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeNormalDecisionTree()
306 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeObliviousDecisionTree()
439 const auto& partition_ids = partition_ids_t->vec<int32>(); in Compute() local
714 const auto& partition_ids = partition_ids_t->vec<int32>(); in Compute() local
816 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeNormalDecisionTree()
889 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeObliviousDecisionTree()
|
| D | stats_accumulator_ops.cc | 140 auto partition_ids = partition_ids_t->vec<int32>(); in SerializeScalarAccumulatorToOutput() local
181 auto partition_ids = partition_ids_t->vec<int32>(); in SerializeTensorAccumulatorToOutput() local
229 const auto& partition_ids = partition_ids_t.vec<int32>(); in AddToScalarAccumulator() local
274 const auto& partition_ids = partition_ids_t.vec<int32>(); in AddToTensorAccumulator() local
|
| D | training_ops.cc | 448 const auto& partition_ids = partition_ids_list[handler_id].vec<int32>(); in FindBestSplitsPerPartitionNormal() local
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 86 partition_ids, argument
|