| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/accumulators/ |
| class-partition-key.h | 28 ClassPartitionKey() : class_id(-1), partition_id(-1), feature_id(-1) {}
31 : class_id(c), partition_id(p), feature_id(f) {}
34 return (feature_id == other.feature_id) &&
43 return Hash64Combine(class_partition, key.feature_id);
54 uint64 feature_id; member in struct:tensorflow::boosted_trees::learner::ClassPartitionKey
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| split_collection_operators.cc | 118 decision_trees::FeatureId feature_id; local
119 input_data->RandomSample(example, &feature_id, &bias, &type);
124 *test->mutable_feature_id() = feature_id;
129 *test.mutable_feature_id() = feature_id;
|
| grow_stats_test.cc | 46 FeatureId feature_id; local
47 feature_id.mutable_id()->set_value(feat);
48 *test->mutable_feature_id() = feature_id;
|
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| extract_feature.cc | 94 int64_t feature_id = local
96 label->data.i32[i] = static_cast<int32_t>(feature_id);
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| examples_iterable.h | 126 const int32 feature_id = indices(row_idx, 1); local
131 feature_id,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| stats_accumulator_ops.cc | 42 PartitionKey() : partition_id(-1), feature_id(-1), dimension(-1) {}
45 : partition_id(p), feature_id(f), dimension(d) {}
49 (dimension == other.dimension) && (feature_id == other.feature_id);
62 (a.feature_id < b.feature_id)) {
73 int64 feature_id; member in struct:tensorflow::boosted_trees::__anon39136::PartitionKey
164 feature_ids(i, 0) = iter.first.feature_id;
209 feature_ids(i, 0) = iter.first.feature_id;
[all...] |
