Searched refs:feature_t (Results 1 – 1 of 1) sorted by relevance
1335 const Tensor* feature_t; in Compute() local1336 OP_REQUIRES_OK(context, context->input("feature", &feature_t)); in Compute()1337 const auto feature = feature_t->matrix<int32>(); in Compute()1344 const int64 feature_dims = feature_t->dim_size(1); in Compute()