Searched refs:feature_indices_t (Results 1 – 1 of 1) sorted by relevance
1540 const Tensor* feature_indices_t; in Compute() local1542 context->input("feature_indices", &feature_indices_t)); in Compute()1543 const auto feature_indices = feature_indices_t->matrix<int32>(); in Compute()1564 const int64 num_sparse_entries = feature_indices_t->dim_size(0); in Compute()