Searched defs:feature_counts (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_cross_op.cc | 54 SparseTensorColumn(const Tensor& values, std::vector<int64> feature_counts, argument
57 feature_counts_(std::move(feature_counts)),
454 std::vector<std::vector<int64>> feature_counts(number_of_columns,
459 ExtractFeatureData(indices_list_in, batch_size, &feature_counts,
465 values_list_in[i], std::move(feature_counts[i]),
479 std::vector<std::vector<int64>>* feature_counts,
493 (*feature_counts)[i].push_back(feature_count);
477 ExtractFeatureData( const OpInputList& indices_list_in, int64 batch_size, std::vector<std::vector<int64>>* feature_counts, std::vector<std::vector<int64>>* feature_start_indices) argument
|
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| H A D | sparse_feature_cross_kernel.cc | 54 SparseTensorColumn(const Tensor& values, std::vector<int64> feature_counts, argument
57 feature_counts_(std::move(feature_counts)),
497 std::vector<std::vector<int64>> feature_counts(number_of_columns,
502 ExtractFeatureData(indices_list_in, batch_size, &feature_counts,
508 values_list_in[i], std::move(feature_counts[i]),
522 std::vector<std::vector<int64>>* feature_counts,
536 (*feature_counts)[i].push_back(feature_count);
520 ExtractFeatureData( const OpInputList& indices_list_in, int64 batch_size, std::vector<std::vector<int64>>* feature_counts, std::vector<std::vector<int64>>* feature_start_indices) argument
|
Completed in 131 milliseconds