Searched refs:tree_biases (Results 1 - 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| H A D | hard_routing_function_op.cc | 54 .Input("tree_biases: float")
79 tree_biases: `tree_biases[i]` gives the bias of the logistic
135 const auto tree_biases = tree_biases_tensor.tensor<float, 1>(); variable
147 tree_biases(j), num_features);
156 if (dot_product < tree_biases(j)) {
|
| H A D | routing_function_op.cc | 53 .Input("tree_biases: float")
73 tree_biases: `tree_biases[i]` gives the bias of the logistic
114 const auto tree_biases = tree_biases_tensor.tensor<float, 1>(); variable
129 tree_biases(j), num_features);
|
| H A D | routing_gradient_op.cc | 48 .Input("tree_biases: float")
68 tree_biases: `tree_biases[i]` gives the bias of the logistic
112 const auto tree_biases = tree_biases_tensor.tensor<float, 1>(); variable
129 tree_biases(j), num_features);
|
| H A D | k_feature_gradient_op.cc | 43 .Input("tree_biases: float")
61 tree_biases: `tree_biases[i]` gives the bias of the logistic
139 const auto tree_biases = tree_biases_tensor.tensor<float, 1>(); variable
166 tree_biases(j), num_features, num_features_per_node);
|
| H A D | k_feature_routing_function_op.cc | 56 .Input("tree_biases: float")
82 tree_biases: `tree_biases[i]` gives the bias of the logistic
129 const auto tree_biases = tree_biases_tensor.tensor<float, 1>(); variable
149 tree_biases(j), num_features, num_features_per_node_);
|
| H A D | stochastic_hard_routing_function_op.cc | 58 .Input("tree_biases: float")
84 tree_biases: `tree_biases[i]` gives the bias of the logistic
145 const auto tree_biases = tree_biases_tensor.tensor<float, 1>(); variable
161 tree_biases(j), num_features);
|
| H A D | stochastic_hard_routing_gradient_op.cc | 48 .Input("tree_biases: float")
81 tree_biases: `tree_biases[i]` gives the bias of the logistic
173 const auto tree_biases = tree_biases_tensor.tensor<float, 1>(); variable
205 tree_biases(j), num_features);
|
Completed in 139 milliseconds