Searched refs:IteratorResource (Results 1 – 2 of 2) sorted by relevance
79 IteratorResource::IteratorResource( in IteratorResource() function in tensorflow::data::IteratorResource100 IteratorResource::~IteratorResource() { in ~IteratorResource()104 Status IteratorResource::GetNext(OpKernelContext* ctx, in GetNext()166 Status IteratorResource::Save(SerializationContext* ctx, in Save()183 Status IteratorResource::Restore(OpKernelContext* ctx, in Restore()229 Status IteratorResource::SetIteratorFromDataset(OpKernelContext* ctx, in SetIteratorFromDataset()371 IteratorResource* iterator_resource) { in InitializeFromIterator()460 ->template Delete<IteratorResource>(cinfo_.container(), in ~IteratorHandleOp()492 IteratorResource* resource; in Compute()495 mgr->LookupOrCreate<IteratorResource>( in Compute()[all …]
33 class IteratorResource : public ResourceBase {35 IteratorResource(Env* env, const DataTypeVector& output_dtypes,42 ~IteratorResource() override;157 Status VerifyResource(IteratorResource* resource);166 IteratorResource* resource_ TF_GUARDED_BY(mu_) = nullptr;177 class AnonymousIteratorHandleOp : public AnonymousResourceOp<IteratorResource> {188 IteratorResource** resource) override;