Searched refs:RecordYielder (Results 1 – 3 of 3) sorted by relevance
58 class RecordYielder {85 explicit RecordYielder(OpKernelConstruction* context,86 const RecordYielder::Options& opts);87 ~RecordYielder();89 RecordYielder(const RecordYielder&) = delete;90 RecordYielder& operator=(const RecordYielder&) = delete;99 typedef RecordYielder ME;
24 RecordYielder::RecordYielder(OpKernelConstruction* context, in RecordYielder() function in tensorflow::RecordYielder25 const RecordYielder::Options& opts) in RecordYielder()35 RecordYielder::~RecordYielder() { in ~RecordYielder()47 Status RecordYielder::YieldOne(string* value) { in YieldOne()71 struct RecordYielder::Shard {78 bool RecordYielder::ShouldFinish(const Status& s) { in ShouldFinish()97 void RecordYielder::MainLoop() { in MainLoop()174 bool RecordYielder::Add(std::vector<string>* values) { in Add()198 void RecordYielder::ShardLoop(Shard* shard) { in ShardLoop()
44 RecordYielder::Options yopts; in RecordInputOp()51 yielder_ = std::unique_ptr<RecordYielder>(new RecordYielder(ctx, yopts)); in RecordInputOp()67 std::unique_ptr<RecordYielder> yielder_;