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

/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
H A Dhard_routing_function_op.cc114 Tensor* output_probability = nullptr; variable
126 &output_probability));
130 auto out_probability = output_probability->tensor<float, 2>();
H A Dstochastic_hard_routing_function_op.cc127 Tensor* output_probability = nullptr; variable
139 &output_probability));
143 auto out_probability = output_probability->tensor<float, 2>();

Completed in 116 milliseconds