Searched refs:shard_ (Results 1 – 3 of 3) sorted by relevance
173 shard_(shard), in Dataset()203 TF_RETURN_IF_ERROR(b->AddScalar(shard_, &shard)); in AsGraphDefInternal()294 if (dataset()->shard_ == "") { in SetupStreamsLocked()315 if (entry.GetShardId() == dataset()->shard_.c_str()) { in SetupStreamsLocked()323 return errors::InvalidArgument("no shard ", dataset()->shard_, in SetupStreamsLocked()351 const std::string shard_; member in tensorflow::KinesisDatasetOp::Dataset
240 shard_(shard) { in FixedUnigramSampler()256 shard_(shard) { in FixedUnigramSampler()277 if (word_id % num_shards_ == shard_) weights_.push_back(0.0); in FillReservedIds()295 if (word_id % num_shards_ == shard_) { in LoadFromFile()315 if (word_id % num_shards_ == shard_) { in LoadFromUnigrams()
239 int32 shard_; variable