Home
last modified time | relevance | path

Searched refs:summary_indices (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dstats_ops_test.py132 summary_indices = np.asarray(slices)
133 summary_indices = np.moveaxis(summary_indices, 0, 1)
140 summary_indices,
156 (summary_indices, summary_values,
163 summary_indices,
548 (summary_indices, summary_values,
555 summary_indices,
717 (summary_indices, summary_values,
724 summary_indices,
883 (summary_indices, summary_values,
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc1613 auto summary_indices = summary_indices_t->matrix<int32>(); in Compute() local
1622 summary_indices(entry_index, 0) = iter.first.node_id; in Compute()
1623 summary_indices(entry_index, 1) = iter.first.feature_dim; in Compute()
1624 summary_indices(entry_index, 2) = iter.first.bucket_id; in Compute()
1625 summary_indices(entry_index, 3) = stat_dim; in Compute()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BoostedTreesSparseCalculateBestFeatureSplit.pbtxt20 …fies the number of non-zero values), which supplies the values for each element in summary_indices.