| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
| D | ordinal_split_handler.py | 152 def reset(self, stamp_token, next_stamp_token): argument
153 reset_1 = self._stats_accumulator.flush(stamp_token, next_stamp_token)
154 reset_2 = self._quantile_accumulator.flush(stamp_token, next_stamp_token)
262 def make_splits(self, stamp_token, next_stamp_token, class_id): argument
273 next_stamp_token, self._multiclass_strategy, class_id,
282 stamp_token, next_stamp_token, multiclass_strategy, argument
304 next_stamp_token=next_stamp_token)
309 stats_accumulator_handle, stamp_token, next_stamp_token))
313 stats_accumulator_handle, stamp_token, next_stamp_token))
442 def make_splits(self, stamp_token, next_stamp_token, class_id): argument
[all …]
|
| D | categorical_split_handler.py | 186 def make_splits(self, stamp_token, next_stamp_token, class_id): argument
191 self._stats_accumulator.flush(stamp_token, next_stamp_token))
219 def reset(self, stamp_token, next_stamp_token): argument
220 reset = self._stats_accumulator.flush(stamp_token, next_stamp_token)
|
| D | base_split_handler.py | 137 def reset(self, stamp_token, next_stamp_token): argument
141 def make_splits(self, stamp_token, next_stamp_token, class_id): argument
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | quantile_ops.py | 213 def flush(self, stamp_token, next_stamp_token): argument
225 next_stamp_token=next_stamp_token)
227 def flush_summary(self, stamp_token, next_stamp_token): argument
232 next_stamp_token=next_stamp_token)
|
| D | stats_accumulator_ops.py | 238 def flush(self, stamp_token, next_stamp_token): argument
242 self.resource_handle, stamp_token, next_stamp_token)
245 self.resource_handle, stamp_token, next_stamp_token)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | stats_accumulator_ops_test.py | 48 stamp_token=0, next_stamp_token=1)
76 stamp_token=0, next_stamp_token=1)
110 stamp_token=0, next_stamp_token=1)
140 accumulator.flush(stamp_token=0, next_stamp_token=1))
187 stamp_token=2, next_stamp_token=3)
245 stamp_token=0, next_stamp_token=1)
283 stamp_token=0, next_stamp_token=1)
321 stamp_token=0, next_stamp_token=1)
356 accumulator.flush(stamp_token=0, next_stamp_token=1))
414 stamp_token=2, next_stamp_token=3)
|
| D | training_ops_test.py | 181 next_stamp_token=1,
232 next_stamp_token=2,
279 next_stamp_token=3,
342 next_stamp_token=1,
456 next_stamp_token=1,
608 next_stamp_token=1,
794 next_stamp_token=1,
934 next_stamp_token=1,
1004 next_stamp_token=1,
1109 next_stamp_token=1,
[all …]
|
| D | quantile_ops_test.py | 131 sess.run(accumulator.flush(stamp_token=0, next_stamp_token=1))
178 sess.run(accumulator.flush(stamp_token=0, next_stamp_token=1))
210 sess.run(accumulator.flush(stamp_token=0, next_stamp_token=1))
288 sess.run(accumulator.flush(stamp_token=0, next_stamp_token=1))
327 accumulator.flush_summary(stamp_token=0, next_stamp_token=1))
331 sess.run(accumulator_2.flush(stamp_token=0, next_stamp_token=1))
362 reset = accumulator.flush(stamp_token=0, next_stamp_token=1)
378 reset = accumulator.flush(stamp_token=0, next_stamp_token=1)
408 reset = accumulator.flush(stamp_token=0, next_stamp_token=1)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 570 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute() local
571 CHECK(stamp_token != next_stamp_token); in Compute()
581 accumulator_resource->set_stamp(next_stamp_token); in Compute()
607 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute() local
613 CHECK(stamp_token != next_stamp_token); in Compute()
621 accumulator_resource->set_stamp(next_stamp_token); in Compute()
|
| D | training_ops.cc | 215 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute() local
216 CHECK(stamp_token != next_stamp_token); in Compute()
219 ensemble_resource->set_stamp(next_stamp_token); in Compute()
341 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute() local
342 CHECK(stamp_token != next_stamp_token); in Compute()
346 ensemble_resource->set_stamp(next_stamp_token); in Compute()
|
| D | quantile_ops.cc | 609 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute() local
628 streams_resource->Reset(next_stamp_token); in Compute()
655 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute() local
677 streams_resource->Reset(next_stamp_token); in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 950 next_stamp_token = stamp_token + 1
954 reset_ops.append(handler.reset(stamp_token, next_stamp_token))
957 bias_stats_accumulator.flush(stamp_token, next_stamp_token))
958 reset_ops.append(steps_accumulator.flush(stamp_token, next_stamp_token))
964 stamp_token=next_stamp_token,
1090 next_stamp_token=next_ensemble_stamp,
1111 next_stamp_token=next_ensemble_stamp,
1127 next_stamp_token=next_ensemble_stamp,
|