Searched refs:prefetch_input_elements_ (Results 1 – 4 of 4) sorted by relevance
43 prefetch_input_elements_(prefetch_input_elements), in ParallelInterleaveDatasetParams()65 CreateTensor<int64_t>(TensorShape({}), {prefetch_input_elements_})); in GetInputTensors()106 int64_t prefetch_input_elements_; member in tensorflow::data::experimental::__anon0cf95fbd0111::ParallelInterleaveDatasetParams
110 prefetch_input_elements_(prefetch_input_elements), in Dataset()202 b->AddScalar(prefetch_input_elements_, &prefetch_input_elements_node)); in AsGraphDefInternal()227 return cycle_length_ + prefetch_input_elements_; in num_threads()705 DCHECK(staging_indices_.size() == dataset()->prefetch_input_elements_); in EnsureWorkerThreadsStarted()1210 const int64_t prefetch_input_elements_; member in tensorflow::data::experimental::ParallelInterleaveDatasetOp::Dataset
212 prefetch_input_elements_(ComputePrefetchInputElements( in Dataset()315 TF_RETURN_IF_ERROR(b->AddScalar(prefetch_input_elements_, in AsGraphDefInternal()380 dataset()->prefetch_input_elements_ + dataset()->cycle_length_; in Initialize()466 ComputeMaxBufferedElements(dataset()->prefetch_input_elements_, in CreateNode()846 dataset()->prefetch_input_elements_ + dataset()->cycle_length_; in WorkerManagerThread()990 dataset()->prefetch_input_elements_ || in FutureWorkerThread()1670 const int64_t prefetch_input_elements_; member in tensorflow::data::ParallelInterleaveDatasetOp::Dataset
46 prefetch_input_elements_(prefetch_input_elements), in ParallelInterleaveDatasetParams()70 CreateTensor<int64_t>(TensorShape({}), {prefetch_input_elements_})); in GetInputTensors()113 int64_t prefetch_input_elements_; member in tensorflow::data::__anonf2f0f7400111::ParallelInterleaveDatasetParams