Searched refs:tree_biases_tensor (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
H A Drouting_function_op.cc90 const Tensor& tree_biases_tensor = context->input(2); variable
114 const auto tree_biases = tree_biases_tensor.tensor<float, 1>();
H A Drouting_gradient_op.cc95 const Tensor& tree_biases_tensor = context->input(2); variable
112 const auto tree_biases = tree_biases_tensor.tensor<float, 1>();
H A Dhard_routing_function_op.cc99 const Tensor& tree_biases_tensor = context->input(2); variable
135 const auto tree_biases = tree_biases_tensor.tensor<float, 1>();
H A Dk_feature_gradient_op.cc97 const Tensor& tree_biases_tensor = context->input(2); variable
139 const auto tree_biases = tree_biases_tensor.tensor<float, 1>();
H A Dk_feature_routing_function_op.cc105 const Tensor& tree_biases_tensor = context->input(2); variable
129 const auto tree_biases = tree_biases_tensor.tensor<float, 1>();
H A Dstochastic_hard_routing_function_op.cc110 const Tensor& tree_biases_tensor = context->input(2); variable
145 const auto tree_biases = tree_biases_tensor.tensor<float, 1>();
H A Dstochastic_hard_routing_gradient_op.cc121 const Tensor& tree_biases_tensor = context->input(2); variable
173 const auto tree_biases = tree_biases_tensor.tensor<float, 1>();

Completed in 113 milliseconds