Searched refs:feature_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 255 const int32 feature_dims = stats_summary_t->dim_size(1); in Compute() local
322 stats_summary, node_id, feature_dims, logits_dim, hessian_dim, in Compute()
328 stats_summary, total_grad, total_hess, node_id, feature_dims, in Compute()
425 const int32 feature_dims, const int32 logits_dim, const int32 hessian_dim, in CalculateBestInequalitySplit() argument
436 for (int f_dim = 0; f_dim < feature_dims; ++f_dim) { in CalculateBestInequalitySplit()
494 const int32 node_id, const int32 feature_dims, const int32 logits_dim, in CalculateBestEqualitySplit() argument
502 for (int f_dim = 0; f_dim < feature_dims; ++f_dim) { in CalculateBestEqualitySplit()
581 const int32 feature_dims = stats_summaries_list[0].dim_size(1); in Compute() local
689 stats_summary, node_id, feature_dims, logits_dim, hessian_dim, in Compute()
695 stats_summary, total_grad, total_hess, node_id, feature_dims, in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 1521 batch_size, feature_dims, argument
1528 dense_indices = np.random.randint(2, size=(batch_size, feature_dims))
1532 feature_shape = np.asarray([batch_size, feature_dims])
1535 (max_splits, feature_dims, num_buckets + 1, stats_dims))
1553 feature_dims = 1
1560 max_splits, num_buckets, batch_size, feature_dims, logits_dims,
1575 feature_dims = 1
1582 max_splits, num_buckets, batch_size, feature_dims, logits_dims,
1597 feature_dims = 1
1604 max_splits, num_buckets, batch_size, feature_dims, logits_dims,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesCalculateBestFeatureSplit.pbtxt | 13 A Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated stats summa…
|
| D | api_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt | 13 A list of Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated sta…
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 1646 feature_dims = (3, 4, 5)
1647 input_size = feature_dims
1662 cell = DummyMultiDimensionalLSTM(feature_dims)
|