| /external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
| D | weighted_quantiles_summary.h | 48 max_rank = max; in SummaryEntry()
57 max_rank = 0; in SummaryEntry()
62 min_rank == other.min_rank && max_rank == other.max_rank;
67 << entry.min_rank << ", " << entry.max_rank << "}";
71 WeightType PrevMaxRank() const { return max_rank - weight; } in PrevMaxRank()
79 WeightType max_rank; member
146 it1->max_rank + it2->PrevMaxRank()); in Merge()
155 it2->max_rank + it1->PrevMaxRank()); in Merge()
163 it1->max_rank + it2->max_rank); in Merge()
176 it1->max_rank + other_entries.back().max_rank); in Merge()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/ |
| D | weighted_quantiles_summary.h | 42 max_rank = max; in SummaryEntry()
49 max_rank = 0; in SummaryEntry()
54 min_rank == other.min_rank && max_rank == other.max_rank;
59 << entry.min_rank << ", " << entry.max_rank << "}";
63 WeightType PrevMaxRank() const { return max_rank - weight; } in PrevMaxRank()
71 WeightType max_rank; member
138 it1->max_rank + it2->PrevMaxRank()); in Merge()
147 it2->max_rank + it1->PrevMaxRank()); in Merge()
155 it1->max_rank + it2->max_rank); in Merge()
168 it1->max_rank + other_entries.back().max_rank); in Merge()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.cc | 69 int max_rank = std::max(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
71 if (min_rank != max_rank) { in Compile()
75 extend_dimension.push_back(max_rank - min_rank + i); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 49 int64 max_rank = -1; in LayoutsAreReduceInputFusionFriendly() local
52 if (param->shape().IsArray() && param->shape().rank() > max_rank) { in LayoutsAreReduceInputFusionFriendly()
53 max_rank = param->shape().rank(); in LayoutsAreReduceInputFusionFriendly()
58 return (!param->shape().IsArray()) || (param->shape().rank() < max_rank) || in LayoutsAreReduceInputFusionFriendly()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | cwise_op_vectorizer.cc | 69 Output max_rank = ops::Maximum(scope, rank_0, rank_1); in ExpandDimsForBroadcast() local
73 [&scope, &max_rank](const WrappedTensor& tensor, const Output& rank) { in ExpandDimsForBroadcast()
79 Output rank_diff = ops::Sub(scope, max_rank, rank); in ExpandDimsForBroadcast()
|
| /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 | 23 float max_rank = 4; field
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 197 output(row, 3) = entry.max_rank; in Compute()
254 float max_rank = summary_values(i, 3); in Compute() local
255 QuantileSummaryEntry entry(value, weight, min_rank, max_rank); in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 216 new_entry->set_max_rank(entry.max_rank); in CopySummaryToProto()
333 entry.min_rank(), entry.max_rank()); in Compute()
558 entry.max_rank()); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 709 max_rank = ranks[0]
711 max_rank = math_ops.maximum(rank, max_rank)
716 rank_diff = array_ops.reshape(max_rank - ranks[i], [1])
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1905 const int64 max_rank = 4; in PropagateShapes() local
1907 max_rank * max_loop_length * std::max<int64>(1, num_loops * num_loops); in PropagateShapes()
1909 const int64 max_resource_iterations = num_queues * num_queues * max_rank; in PropagateShapes()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 296 int max_rank = kDefaultMaxRank,
443 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims() argument
446 CHECK_LE(min_rank, max_rank); in RandomDims()
447 std::uniform_int_distribution<int> rank_distribution(min_rank, max_rank); in RandomDims()
|