Searched refs:sparse_float_feature_indices_list (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
| D | batch_features_testutil.cc | 45 std::vector<tensorflow::Tensor> sparse_float_feature_indices_list; in RandomlyInitializeBatchFeatures() local
64 sparse_float_feature_indices_list.push_back(indices_tensor); in RandomlyInitializeBatchFeatures()
81 dense_float_features_list, sparse_float_feature_indices_list, in RandomlyInitializeBatchFeatures()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | prediction_ops.cc | 196 OpInputList sparse_float_feature_indices_list; in DoCompute() local
200 context, &sparse_float_feature_indices_list, in DoCompute()
225 sparse_float_feature_indices_list), in DoCompute()
408 OpInputList sparse_float_feature_indices_list; in DoCompute() local
412 context, &sparse_float_feature_indices_list, in DoCompute()
437 sparse_float_feature_indices_list), in DoCompute()
|
| D | quantile_ops.cc | 365 OpInputList sparse_float_feature_indices_list; in Compute() local
369 context, &sparse_float_feature_indices_list, in Compute()
425 sparse_float_feature_indices_list[sparse_index].matrix<int64>(); in Compute()
434 sparse_float_feature_indices_list[sparse_index].dim_size(0); in Compute()
779 OpInputList sparse_float_feature_indices_list; in Compute() local
783 context, &sparse_float_feature_indices_list, in Compute()
810 sparse_float_feature_indices_list[sparse_index].matrix<int64>(); in Compute()
815 sparse_float_feature_indices_list[sparse_index].dim_size(0); in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | batch_features.cc | 27 std::vector<Tensor> sparse_float_feature_indices_list, in Initialize() argument
35 auto num_sparse_float_features = sparse_float_feature_indices_list.size(); in Initialize()
72 sparse_float_feature_indices_list[sparse_feat_idx]; in Initialize()
|
| D | batch_features.h | 41 std::vector<Tensor> sparse_float_feature_indices_list,
|