| /external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
| D | weighted_quantiles_summary.h | 47 min_rank = min; in SummaryEntry()
56 min_rank = 0; in SummaryEntry()
62 min_rank == other.min_rank && max_rank == other.max_rank;
67 << entry.min_rank << ", " << entry.max_rank << "}";
74 WeightType NextMinRank() const { return min_rank + weight; } in NextMinRank()
78 WeightType min_rank; member
145 it1->min_rank + next_min_rank2, in Merge()
154 it2->min_rank + next_min_rank1, in Merge()
162 it1->min_rank + it2->min_rank, in Merge()
175 it1->min_rank + next_min_rank2, in Merge()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/ |
| D | weighted_quantiles_summary.h | 41 min_rank = min; in SummaryEntry()
48 min_rank = 0; in SummaryEntry()
54 min_rank == other.min_rank && max_rank == other.max_rank;
59 << entry.min_rank << ", " << entry.max_rank << "}";
66 WeightType NextMinRank() const { return min_rank + weight; } in NextMinRank()
70 WeightType min_rank; member
137 it1->min_rank + next_min_rank2, in Merge()
146 it2->min_rank + next_min_rank1, in Merge()
154 it1->min_rank + it2->min_rank, in Merge()
167 it1->min_rank + next_min_rank2, in Merge()
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | utils.py | 228 def _get_dimension(shape, dim, min_rank=1): argument
246 if len(dims) < min_rank:
247 raise ValueError('rank of shape must be at least %d not: %d' % (min_rank,
256 def channel_dimension(shape, data_format, min_rank=1): argument
272 min_rank=min_rank)
275 def last_dimension(shape, min_rank=1): argument
289 return _get_dimension(shape, -1, min_rank=min_rank)
|
| D | layers.py | 1269 num_filters_in = utils.last_dimension(inputs.get_shape(), min_rank=4)
2800 inputs.get_shape(), df, min_rank=4)
2902 num_logits = utils.last_dimension(logits.get_shape(), min_rank=2)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.cc | 70 int min_rank = std::min(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
71 if (min_rank != max_rank) { in Compile()
72 for (int i = 0; i < min_rank; ++i) { in Compile()
75 extend_dimension.push_back(max_rank - min_rank + i); in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesQuantileStreamResourceAddSummaries.pbtxt | 19 summary Tensor is rank 2, containing summaries (value, weight, min_rank, max_rank)
|
| D | api_def_BoostedTreesMakeQuantileSummaries.pbtxt | 26 (value, weight, min_rank, max_rank) of a single feature.
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/proto/ |
| D | quantiles.proto | 22 float min_rank = 3; field
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 196 output(row, 2) = entry.min_rank; in Compute()
253 float min_rank = summary_values(i, 2); in Compute() local
255 QuantileSummaryEntry entry(value, weight, min_rank, max_rank); in Compute()
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| D | resnet_v1.py | 105 depth_in = utils.last_dimension(inputs.get_shape(), min_rank=4)
|
| D | resnet_v2.py | 99 depth_in = utils.last_dimension(inputs.get_shape(), min_rank=4)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 215 new_entry->set_min_rank(entry.min_rank); in CopySummaryToProto()
333 entry.min_rank(), entry.max_rank()); in Compute()
557 entries.emplace_back(entry.value(), entry.weight(), entry.min_rank(), in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1253 const int min_rank = concat_dim < 0 ? -concat_dim : concat_dim + 1; in ConcatShapeHelper() local
1259 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, min_rank, &input)); in ConcatShapeHelper()
1272 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, min_rank, &input)); in ConcatShapeHelper()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 295 std::vector<int64> RandomDims(int min_rank = 0,
443 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims() argument
445 CHECK_LE(0, min_rank); in RandomDims()
446 CHECK_LE(min_rank, max_rank); in RandomDims()
447 std::uniform_int_distribution<int> rank_distribution(min_rank, max_rank); in RandomDims()
|