Searched defs:feature_set (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
H A D | k_feature_routing_function_op.cc | 132 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 D | utils.cc | 42 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 D | k_feature_gradient_op.cc | 146 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