Searched defs:feature_set (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
H A Dk_feature_routing_function_op.cc132 std::vector<int32> feature_set; variable
139 feature_set.clear();
141 num_features_per_node_, &feature_set);
148 point, feature_set, tree_parameters_tensor.Slice(j, j + 1),
H A Dutils.cc42 float LeftProbabilityK(const Tensor& point, std::vector<int32> feature_set, argument
51 CHECK_LT(feature_set[i], num_features);
52 dot_product += p(feature_set[i]) * w(i);
H A Dk_feature_gradient_op.cc146 std::vector<int32> feature_set; variable
149 feature_set.clear();
154 num_features_per_node, &feature_set);
165 point, feature_set, tree_parameters_tensor.Slice(j, j + 1),
175 CHECK_LT(feature_set[k], num_features);
176 data_grad(j, feature_set[k]) = tree_parameters(j, k);
177 weights_grad(i, j, k) = input_data(i, feature_set[k]);

Completed in 104 milliseconds