Searched refs:columns_tensor (Results 1 – 1 of 1) sorted by relevance
200 const Tensor* columns_tensor; in ComputeAsync() local201 OP_REQUIRES_OK_ASYNC(ctx, ctx->input("columns", &columns_tensor), done); in ComputeAsync()202 OP_REQUIRES_ASYNC(ctx, columns_tensor->dims() == 1, in ComputeAsync()207 columns_tensor->NumElements() == in ComputeAsync()218 columns.push_back(columns_tensor->flat<string>()(i)); in ComputeAsync()