Searched refs:num_outputs_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.h | 110 const int32 num_outputs_; variable
306 total_counts_.resize(num_outputs_); in Initialize()
317 left_counts_.resize(num_outputs_ * num_splits()); in ClassificationAddSplitStats()
320 left_counts_.erase(left_counts_.begin() + num_outputs_ * split_num, in ClassificationRemoveSplitStats()
321 left_counts_.begin() + num_outputs_ * (split_num + 1)); in ClassificationRemoveSplitStats()
343 return left_counts_[split * num_outputs_ + class_num]; in left_count()
347 left_counts_[split * num_outputs_ + class_num]; in right_count()
352 return left_counts_[split * num_outputs_ + class_num]; in mutable_left_count()
534 total_sum_.resize(num_outputs_); in Initialize()
535 total_sum_squares_.resize(num_outputs_); in Initialize()
[all …]
|
| D | grow_stats.cc | 39 num_outputs_(params.num_outputs()) {} in GrowStats()
172 *left_sum, left_gini_->square(split), num_outputs_); in MaybeCachedGiniScore()
176 *right_sum, right_gini_->square(split), num_outputs_); in MaybeCachedGiniScore()
322 float denom = static_cast<float>(n) + static_cast<float>(num_outputs_); in MakeBootstrapWeights()
323 for (int i = 0; i < num_outputs_; ++i) { in MakeBootstrapWeights()
327 (*weights)[num_outputs_ + i] = (right_count(index, i) + 1.0) / denom; in MakeBootstrapWeights()
354 std::vector<float> weights1(num_outputs_ * 2); in CheckFinishEarlyBootstrap()
358 std::vector<float> weights2(num_outputs_ * 2); in CheckFinishEarlyBootstrap()
366 int g1 = BootstrapGini(weight_sum_, 2 * num_outputs_, ds1, rng_.get()); in CheckFinishEarlyBootstrap()
372 int g2 = BootstrapGini(weight_sum_, 2 * num_outputs_, ds2, rng_.get()); in CheckFinishEarlyBootstrap()
[all …]
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 86 int num_outputs_; member in tensorflow::__anon8001287d0111::GenTypeScriptOp
90 : op_def_(op_def), api_def_(api_def), num_outputs_(0) {} in GenTypeScriptOp()
135 num_outputs_ = api_def_.out_arg_size(); in ProcessArgs()
189 if (num_outputs_ == 1) { in AddMethodSignature()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 184 : num_outputs_(num_outputs), in MultiDeviceFunctionData()
191 const int num_outputs_; member
|
| D | process_function_library_runtime.cc | 720 output_devices->resize(data->num_outputs_); in GetOutputDevices()
785 rets->resize(data->num_outputs_); in RunMultiDevice()
|