Searched refs:out_probability (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
H A Dhard_routing_function_op.cc130 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 Dstochastic_hard_routing_function_op.cc143 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