Searched refs:out_probability (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
H A D | hard_routing_function_op.cc | 130 auto out_probability = output_probability->tensor<float, 2>(); variable 142 out_probability(i, 0) = 1.0; 157 out_probability(i, j + 1) = left_prob * out_probability(i, j); 161 out_probability(i, j + 1) = (1.0 - left_prob) * out_probability(i, j);
|
H A D | stochastic_hard_routing_function_op.cc | 143 auto out_probability = output_probability->tensor<float, 2>(); variable 153 out_probability(i, 0) = 1.0; 168 out_probability(i, j + 1) = left_prob * out_probability(i, j); 176 out_probability(i, j + 1) = (1.0 - left_prob) * out_probability(i, j);
|
Completed in 105 milliseconds