Searched refs:SparseTensorColumn (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/layers/kernels/
H A Dsparse_feature_cross_kernel.cc52 class SparseTensorColumn : public ColumnInterface<InternalType> { class in namespace:tensorflow::__anon25858
54 SparseTensorColumn(const Tensor& values, std::vector<int64> feature_counts, function in class:tensorflow::__anon25858::SparseTensorColumn
68 ~SparseTensorColumn() override {}
78 int64 SparseTensorColumn<int64>::Feature(int64 batch, int64 n) const {
87 string SparseTensorColumn<string>::Feature(int64 batch, int64 n) const {
95 StringPiece SparseTensorColumn<StringPiece>::Feature(int64 batch,
507 columns.emplace_back(new SparseTensorColumn<InternalType>(
/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_cross_op.cc52 class SparseTensorColumn : public ColumnInterface<InternalType> { class in namespace:tensorflow::__anon26603
54 SparseTensorColumn(const Tensor& values, std::vector<int64> feature_counts, function in class:tensorflow::__anon26603::SparseTensorColumn
68 ~SparseTensorColumn() override {}
78 int64 SparseTensorColumn<int64>::Feature(int64 batch, int64 n) const {
87 string SparseTensorColumn<string>::Feature(int64 batch, int64 n) const {
95 StringPiece SparseTensorColumn<StringPiece>::Feature(int64 batch,
464 columns.emplace_back(new SparseTensorColumn<InternalType>(

Completed in 109 milliseconds