Searched refs:oblivious_split_info (Results 1 – 3 of 3) sorted by relevance
395 ObliviousSplitInfo oblivious_split_info; in ComputeObliviousDecisionTree() local397 oblivious_split_info.mutable_split_node() in ComputeObliviousDecisionTree()404 auto* left_child = oblivious_split_info.add_children(); in ComputeObliviousDecisionTree()405 auto* right_child = oblivious_split_info.add_children(); in ComputeObliviousDecisionTree()412 oblivious_split_info.add_children_parent_id(partition_ids(start_index)); in ComputeObliviousDecisionTree()414 oblivious_split_info.SerializeToString(&(*output_splits)(0)); in ComputeObliviousDecisionTree()976 ObliviousSplitInfo oblivious_split_info; in ComputeObliviousDecisionTree() local978 oblivious_split_info.mutable_split_node() in ComputeObliviousDecisionTree()985 auto* left_child = oblivious_split_info.add_children(); in ComputeObliviousDecisionTree()986 auto* right_child = oblivious_split_info.add_children(); in ComputeObliviousDecisionTree()[all …]
51 learner::ObliviousSplitInfo oblivious_split_info; member500 context, split.oblivious_split_info.ParseFromString(serialized_split), in FindBestSplitOblivious()503 auto split_info = split.oblivious_split_info; in FindBestSplitOblivious()538 trees::TreeNode best_node = best_split.oblivious_split_info.split_node(); in FindBestSplitOblivious()781 auto split_info = split->oblivious_split_info; in SplitTreeLayer()
264 oblivious_split_info = split_info_pb2.ObliviousSplitInfo()265 oblivious_split_info.ParseFromString(splits[0])266 split_node = oblivious_split_info.split_node287 left_child = oblivious_split_info.children[0].vector288 right_child = oblivious_split_info.children[1].vector304 left_child = oblivious_split_info.children[2].vector305 right_child = oblivious_split_info.children[3].vector318 self.assertEqual(2, len(oblivious_split_info.children_parent_id))319 self.assertEqual(1, oblivious_split_info.children_parent_id[0])320 self.assertEqual(2, oblivious_split_info.children_parent_id[1])