Searched refs:row_shape_t (Results 1 – 1 of 1) sorted by relevance
46 const Tensor* row_shape_t; in MakeDataset() local47 OP_REQUIRES_OK(ctx, ctx->input("row_shape", &row_shape_t)); in MakeDataset()48 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(row_shape_t->shape()), in MakeDataset()52 row_shape_t->vec<int64>().data(), in MakeDataset()53 row_shape_t->NumElements(), &row_shape)); in MakeDataset()