Searched refs:param_proto_ (Results 1 – 2 of 2) sorted by relevance
42 ParseProtoUnlimited(¶m_proto_, serialized_params); in CreateTreeVariableOp()51 auto* result = new DecisionTreeResource(param_proto_); in Compute()70 TensorForestParams param_proto_; member in tensorflow::tensorforest::CreateTreeVariableOp99 ParseProtoUnlimited(¶m_proto_, serialized_params); in TreeDeserializeOp()125 TensorForestParams param_proto_; member in tensorflow::tensorforest::TreeDeserializeOp167 ParseProtoUnlimited(¶m_proto_, serialized_params); in TreePredictionsV4Op()172 model_op_ = LeafModelOperatorFactory::CreateLeafModelOperator(param_proto_); in TreePredictionsV4Op()192 const int32 num_outputs = param_proto_.num_outputs(); in Compute()203 param_proto_.inference_tree_paths() ? num_data : 0); in Compute()217 param_proto_.inference_tree_paths() ? &tree_paths : nullptr); in Compute()[all …]
48 ParseProtoUnlimited(¶m_proto_, serialized_params); in CreateFertileStatsVariableOp()56 auto* result = new FertileStatsResource(param_proto_); in Compute()76 TensorForestParams param_proto_; member in tensorflow::tensorforest::CreateFertileStatsVariableOp86 ParseProtoUnlimited(¶m_proto_, serialized_params); in FertileStatsSerializeOp()105 TensorForestParams param_proto_; member in tensorflow::tensorforest::FertileStatsSerializeOp115 ParseProtoUnlimited(¶m_proto_, serialized_params); in FertileStatsDeserializeOp()141 TensorForestParams param_proto_; member in tensorflow::tensorforest::FertileStatsDeserializeOp231 ParseProtoUnlimited(¶m_proto_, serialized_params); in ProcessInputOp()278 if (param_proto_.collate_examples()) { in Compute()298 param_proto_.is_regression() ? (std::max(1, label_dim)) : 1; in Compute()[all …]