Searched refs:sparse_float_feature_indices_list (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
H A Dbatch_features_testutil.cc45 std::vector<tensorflow::Tensor> sparse_float_feature_indices_list; local
64 sparse_float_feature_indices_list.push_back(indices_tensor);
81 dense_float_features_list, sparse_float_feature_indices_list,
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
H A Dbatch_features.cc26 std::vector<Tensor> sparse_float_feature_indices_list,
34 auto num_sparse_float_features = sparse_float_feature_indices_list.size();
71 sparse_float_feature_indices_list[sparse_feat_idx];
24 Initialize( std::vector<Tensor> dense_float_features_list, std::vector<Tensor> sparse_float_feature_indices_list, std::vector<Tensor> sparse_float_feature_values_list, std::vector<Tensor> sparse_float_feature_shapes_list, std::vector<Tensor> sparse_int_feature_indices_list, std::vector<Tensor> sparse_int_feature_values_list, std::vector<Tensor> sparse_int_feature_shapes_list) argument
H A Dbatch_features.h41 std::vector<Tensor> sparse_float_feature_indices_list,
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dprediction_ops.cc188 OpInputList sparse_float_feature_indices_list; local
192 context, &sparse_float_feature_indices_list,
217 sparse_float_feature_indices_list),
370 OpInputList sparse_float_feature_indices_list; local
374 context, &sparse_float_feature_indices_list,
399 sparse_float_feature_indices_list),
H A Dquantile_ops.cc357 OpInputList sparse_float_feature_indices_list; variable
361 context, &sparse_float_feature_indices_list,
417 sparse_float_feature_indices_list[sparse_index].matrix<int64>();
426 sparse_float_feature_indices_list[sparse_index].dim_size(0);
770 OpInputList sparse_float_feature_indices_list; variable
774 context, &sparse_float_feature_indices_list,
801 sparse_float_feature_indices_list[sparse_index].matrix<int64>();
806 sparse_float_feature_indices_list[sparse_index].dim_size(0);

Completed in 162 milliseconds