Searched refs:bigquery_table_accessor_ (Results 1 – 1 of 1) sorted by relevance
57 bigquery_table_accessor_(CHECK_NOTNULL(bigquery_table_accessor)) {} in BigQueryReader()65 TF_RETURN_IF_ERROR(bigquery_table_accessor_->SetPartition(partition)); in OnWorkStartedLocked()73 if (bigquery_table_accessor_->Done()) { in ReadLocked()80 TF_RETURN_IF_ERROR(bigquery_table_accessor_->ReadRow(&row_id, &example)); in ReadLocked()90 BigQueryTableAccessor* bigquery_table_accessor_; member in tensorflow::BigQueryReader111 BigQueryTablePartition(), &bigquery_table_accessor_)); in BigQueryReaderOp()114 return new BigQueryReader(bigquery_table_accessor_.get(), name()); in BigQueryReaderOp()119 std::unique_ptr<BigQueryTableAccessor> bigquery_table_accessor_; member in tensorflow::BigQueryReaderOp143 BigQueryTablePartition(), &bigquery_table_accessor_)); in GenerateBigQueryReaderPartitionsOp()168 total_num_rows_ = bigquery_table_accessor_->total_num_rows(); in InitializeTotalNumberOfRows()[all …]