Searched refs:stats_summaries (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 49 def add_f_dim_and_append_zeros(self, stats_summaries): argument
64 for feature in stats_summaries
214 stats_summaries = self._get_stats_summary_for_split()
215 stats_summaries = self.add_f_dim_and_append_zeros(stats_summaries)
221 stats_summaries,
234 stats_summaries = self._get_stats_summary_for_split()
235 stats_summaries = self.add_f_dim_and_append_zeros(stats_summaries)
241 stats_summaries,
268 stats_summaries = self._get_stats_summary_for_split()
271 stats_summary = np.moveaxis(stats_summaries, 0, 1)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 591 std::vector<TTypes<float, 4>::ConstTensor> stats_summaries; in Compute() local
593 stats_summaries.reserve(num_features_); in Compute()
595 stats_summaries.emplace_back(tensor.tensor<float, 4>()); in Compute()
664 ConstMatrixMap stats_mat(&stats_summaries[0](node_id, 0, 0, 0), in Compute()
679 TTypes<float, 4>::ConstTensor stats_summary = stats_summaries[f_idx]; in Compute()
|