Searched refs:boundaries_vector (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bucketize_op_gpu.cu.cc | 84 const std::vector<float>& boundaries_vector, in Compute()
89 boundaries_vector.size()); in Compute()
91 for (int i = 0; i < boundaries_vector.size(); ++i) { in Compute()
92 boundaries_array.Set(i, boundaries_vector[i]); in Compute()
97 int32 shared_mem_size = sizeof(float) * boundaries_vector.size(); in Compute()
103 d.stream()>>>(input.size(), input.data(), boundaries_vector.size(), in Compute()
109 boundaries_vector.size(), boundaries_array.data(), output.data())); in Compute()
|
| D | bucketize_op.cc | 38 const std::vector<float>& boundaries_vector, in Compute()
43 boundaries_vector.begin(), boundaries_vector.end(), input(i)); in Compute()
44 output(i) = first_bigger_it - boundaries_vector.begin(); in Compute()
|
| D | bucketize_op.h | 34 const std::vector<float>& boundaries_vector,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 940 std::vector<T> boundaries_vector; in Compute() local
941 boundaries_vector.reserve(boundaries.size()); in Compute()
943 boundaries_vector.push_back(boundaries(i)); in Compute()
948 std::is_sorted(boundaries_vector.begin(), boundaries_vector.end()), in Compute()
962 output(i) = CalculateBucketIndex(input(i), boundaries_vector); in Compute()
967 int32 CalculateBucketIndex(const T value, std::vector<T>& boundaries_vector) { in CalculateBucketIndex() argument
968 auto first_bigger_it = std::upper_bound(boundaries_vector.begin(), in CalculateBucketIndex()
969 boundaries_vector.end(), value); in CalculateBucketIndex()
970 int32 index = first_bigger_it - boundaries_vector.begin(); in CalculateBucketIndex()
971 CHECK(index >= 0 && index <= boundaries_vector.size()) in CalculateBucketIndex()
[all …]
|