Searched refs:min_rank (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
| D | weighted_quantiles_summary.h | 48 min_rank = min; in SummaryEntry()
57 min_rank = 0; in SummaryEntry()
63 min_rank == other.min_rank && max_rank == other.max_rank;
68 << entry.min_rank << ", " << entry.max_rank << "}";
75 WeightType NextMinRank() const { return min_rank + weight; } in NextMinRank()
79 WeightType min_rank; member
146 it1->min_rank + next_min_rank2, in Merge()
155 it2->min_rank + next_min_rank1, in Merge()
163 it1->min_rank + it2->min_rank, in Merge()
176 it1->min_rank + next_min_rank2, in Merge()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.cc | 69 int min_rank = std::min(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
70 if (min_rank != max_rank) { in Compile()
71 for (int i = 0; i < min_rank; ++i) { in Compile()
74 extend_dimension.push_back(max_rank - min_rank + i); in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesFlushQuantileSummaries.pbtxt | 13 Each summary Tensor is rank 2, containing summaries (value, weight, min_rank,
|
| 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/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 197 output(row, 2) = entry.min_rank; in Compute()
256 output(row, 2) = entry.min_rank; in Compute()
319 float min_rank = summary_values(i, 2); in Compute() local
321 QuantileSummaryEntry entry(value, weight, min_rank, max_rank); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 296 std::vector<int64> RandomDims(int min_rank = 0,
444 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims() argument
446 CHECK_LE(0, min_rank); in RandomDims()
447 CHECK_LE(min_rank, max_rank); in RandomDims()
448 std::uniform_int_distribution<int> rank_distribution(min_rank, max_rank); in RandomDims()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1932 const int min_rank = concat_dim < 0 ? -concat_dim : concat_dim + 1; in ConcatShapeHelper() local
1938 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, min_rank, &input)); in ConcatShapeHelper()
1951 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, min_rank, &input)); in ConcatShapeHelper()
|