| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | stats_accumulator_ops_test.py | 34 stamp_token=0,
39 stamp_token=0,
48 stamp_token=0, next_stamp_token=1)
62 stamp_token=0,
67 stamp_token=0,
76 stamp_token=0, next_stamp_token=1)
91 stamp_token=0,
96 stamp_token=0,
102 stamp_token=-1,
110 stamp_token=0, next_stamp_token=1)
[all …]
|
| D | training_ops_test.py | 157 stamp_token=0,
180 stamp_token=0,
193 training_ops.tree_ensemble_stats(tree_ensemble_handle, stamp_token=1))
231 stamp_token=1,
242 training_ops.tree_ensemble_stats(tree_ensemble_handle, stamp_token=2))
278 stamp_token=2,
289 training_ops.tree_ensemble_stats(tree_ensemble_handle, stamp_token=3))
308 stamp_token=0,
341 stamp_token=0,
361 training_ops.tree_ensemble_stats(tree_ensemble_handle, stamp_token=1))
[all …]
|
| D | quantile_ops_test.py | 118 stamp_token=0,
131 sess.run(accumulator.flush(stamp_token=0, next_stamp_token=1))
132 are_ready_flush, buckets = (accumulator.get_buckets(stamp_token=1))
153 stamp_token=0,
178 sess.run(accumulator.flush(stamp_token=0, next_stamp_token=1))
179 are_ready_flush, buckets = (accumulator.get_buckets(stamp_token=1))
200 stamp_token=0,
210 sess.run(accumulator.flush(stamp_token=0, next_stamp_token=1))
211 are_ready_flush, buckets = (accumulator.get_buckets(stamp_token=1))
275 stamp_token=0,
[all …]
|
| D | model_ops_test.py | 112 stamp_token=3,
131 stamp_token = model_ops.tree_ensemble_stamp_token(tree_ensemble_handle)
132 self.assertEqual(stamp_token.eval(), 3)
157 stamp_token=7,
160 stamp_token, serialized_config = model_ops.tree_ensemble_serialize(
163 self.assertEqual(stamp_token.eval(), 7)
169 stamp_token=9,
226 stamp_token=3,
241 stamp_token=3,
270 stamp_token=3,
[all …]
|
| D | prediction_ops_test.py | 209 stamp_token=0,
236 stamp_token=0,
266 stamp_token=0,
310 stamp_token=0,
356 stamp_token=0,
432 stamp_token=0,
494 stamp_token=0,
541 stamp_token=0,
591 stamp_token=0,
637 stamp_token=0,
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | quantile_ops.py | 48 stamp_token, state, are_buckets_ready, buckets = (
56 specs = [make_save_spec(stamp_token, "_stamp")]
76 stamp_token = restored_tensors[:1]
83 stamp_token=stamp_token,
158 def get_buckets(self, stamp_token): argument
163 stamp_token=stamp_token))
190 def add_summary(self, stamp_token, column, example_weights): argument
195 stamp_token=stamp_token,
198 def add_prebuilt_summary(self, stamp_token, summary): argument
202 stamp_token=stamp_token,
[all …]
|
| D | stats_accumulator_ops.py | 44 (stamp_token, num_updates, partition_ids, feature_ids, gradients,
47 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
72 def deserialize(self, stamp_token, num_updates, partition_ids, feature_ids, argument
77 self._resource_handle, stamp_token, num_updates, partition_ids,
81 self._resource_handle, stamp_token, num_updates, partition_ids,
97 stamp_token=restored_tensors[0],
115 stamp_token, argument
132 self._stamp_token = stamp_token
196 def add(self, stamp_token, partition_ids, feature_ids, gradients, hessians): argument
202 [self.resource_handle], stamp_token, [partition_ids], [feature_ids],
[all …]
|
| D | model_ops.py | 54 stamp_token, ensemble_config = tree_ensemble_serialize(tree_ensemble_handle)
60 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
84 stamp_token=restored_tensors[0],
91 def __init__(self, stamp_token, tree_ensemble_config, name, container=None): argument
92 self._stamp_token = stamp_token
126 def tree_ensemble_variable(stamp_token, argument
143 tree_ensemble_var = TreeEnsembleVariable(stamp_token, tree_ensemble_config,
|
| /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)
223 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
255 stamp_token=stamp_token,
262 def make_splits(self, stamp_token, next_stamp_token, class_id): argument
272 self._stats_accumulator.resource_handle, stamp_token,
282 stamp_token, next_stamp_token, multiclass_strategy, argument
292 stamp_token=stamp_token))
303 stamp_token=stamp_token,
[all …]
|
| D | base_split_handler.py | 77 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
102 def update_stats_sync(self, stamp_token, example_partition_ids, gradients, argument
126 handler_reads, stamp_token, None)
128 stamp_token, example_partition_ids, gradients, hessians,
133 }, stamp_token, None)
137 def reset(self, stamp_token, next_stamp_token): argument
141 def make_splits(self, stamp_token, next_stamp_token, class_id): argument
|
| D | categorical_split_handler.py | 93 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
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)
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | resource_ops_test.py | 38 stamp_token = ensemble.get_stamp_token()
39 self.assertEqual(0, self.evaluate(stamp_token))
154 stamp_token=7,
157 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
159 self.assertEqual(7, self.evaluate(stamp_token))
169 ensemble = boosted_trees_ops.TreeEnsemble('ensemble', stamp_token=5)
171 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
173 self.assertEqual(5, self.evaluate(stamp_token))
220 stamp_token=3,
223 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch_test.py | 218 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
270 stamp_token, serialized = model_ops.tree_ensemble_serialize(
276 self.assertEquals(stamp_token.eval(), 1)
284 stamp_token, serialized = model_ops.tree_ensemble_serialize(
290 self.assertEquals(stamp_token.eval(), 2)
321 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
363 stamp_token, serialized = model_ops.tree_ensemble_serialize(
369 self.assertEquals(stamp_token.eval(), 1)
373 stamp_token, serialized = model_ops.tree_ensemble_serialize(
379 self.assertEquals(stamp_token.eval(), 2)
[all …]
|
| D | gbdt_batch.py | 574 local_ensemble_handle, stamp_token=-1, tree_ensemble_config="")
590 stamp_token=ensemble_stamp,
830 stamp_token=0,
836 stamp_token=0,
949 stamp_token, _ = model_ops.tree_ensemble_serialize(self._ensemble_handle)
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,
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | boosted_trees_ops.py | 170 stamp_token, serialized = (
177 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
200 stamp_token=restored_tensors[0],
207 def __init__(self, name, stamp_token=0, is_local=False, serialized_proto=''): argument
208 self._stamp_token = stamp_token
253 stamp_token, _, _, _, _ = (
256 return stamp_token
265 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
270 return (array_ops.identity(stamp_token, name='stamp_token'),
287 def deserialize(self, stamp_token, serialized_proto): argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 288 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
295 [&context, &resource_handle_list, &summary_list, stamp_token]( in Compute()
311 if (!streams_resource->is_stamp_valid(stamp_token)) { in Compute()
314 << " Passed stamp token: " << stamp_token << " " in Compute()
337 streams_resource->stream(stamp_token)->PushSummary(entries); in Compute()
478 int64 stamp_token = streams_resource->stamp(); in Compute() local
487 const QuantileStream& stream = *streams_resource->stream(stamp_token); in Compute()
489 are_buckets_ready ? streams_resource->boundaries(stamp_token) in Compute()
511 stamp_token_t->scalar<int64>()() = stamp_token; in Compute()
577 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
[all …]
|
| D | stats_accumulator_ops.cc | 439 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
448 stamp_token](int64 start, int64 end) { in Compute()
462 if (!accumulator_resource->is_stamp_valid(stamp_token)) { in Compute()
464 << "Passed stamp token: " << stamp_token << " " in Compute()
504 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
513 stamp_token](int64 start, int64 end) { in Compute()
527 if (!accumulator_resource->is_stamp_valid(stamp_token)) { in Compute()
529 << "Passed stamp token: " << stamp_token << " " in Compute()
560 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
565 CHECK(accumulator_resource->is_stamp_valid(stamp_token)); in Compute()
[all …]
|
| D | model_ops.cc | 41 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
49 stamp_token)) { in Compute()
124 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
135 tree_ensemble_config_t->scalar<string>()(), stamp_token), in Compute()
159 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
163 CHECK(ensemble_resource->is_stamp_valid(stamp_token)); in Compute()
|
| D | training_ops.cc | 205 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
209 CHECK(ensemble_resource->is_stamp_valid(stamp_token)); in Compute()
216 CHECK(stamp_token != next_stamp_token); in Compute()
331 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
335 CHECK(ensemble_resource->is_stamp_valid(stamp_token)); in Compute()
342 CHECK(stamp_token != next_stamp_token); in Compute()
915 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
919 CHECK(ensemble_resource->is_stamp_valid(stamp_token)); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | resource_ops.cc | 44 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
53 tree_ensemble_serialized_t->scalar<string>()(), stamp_token)) { in Compute()
181 int64 stamp_token = stamp_token_t->scalar<int64>()(); in Compute() local
192 tree_ensemble_serialized_t->scalar<string>()(), stamp_token), in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/resources/ |
| D | quantile_stream_resource.h | 36 int64 stamp_token) in QuantileStreamResource() argument
43 set_stamp(stamp_token); in QuantileStreamResource()
|
| D | decision_tree_ensemble_resource.h | 49 bool InitFromSerialized(const string& serialized, const int64 stamp_token) { in InitFromSerialized() argument
52 set_stamp(stamp_token); in InitFromSerialized()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesSerializeEnsemble.pbtxt | 11 name: "stamp_token"
|
| D | api_def_BoostedTreesDeserializeEnsemble.pbtxt | 11 name: "stamp_token"
|
| D | api_def_BoostedTreesCreateEnsemble.pbtxt | 11 name: "stamp_token"
|