Searched refs:topics_tensor (Results 1 – 1 of 1) sorted by relevance
27 const Tensor* topics_tensor; in MakeDataset() local28 OP_REQUIRES_OK(ctx, ctx->input("topics", &topics_tensor)); in MakeDataset()30 ctx, topics_tensor->dims() <= 1, in MakeDataset()34 topics.reserve(topics_tensor->NumElements()); in MakeDataset()35 for (int i = 0; i < topics_tensor->NumElements(); ++i) { in MakeDataset()36 topics.push_back(topics_tensor->flat<string>()(i)); in MakeDataset()