Searched refs:generator_state (Results 1 – 2 of 2) sorted by relevance
404 generator_state = DatasetV2._GeneratorState(generator)420 generator_state.get_next_id, args, dtypes.int64, stateful=True)440 values = next(generator_state.get_iterator(iterator_id))495 generator_state.iterator_completed(iterator_id)
2882 Node* generator_state = environment()->LookupGeneratorState(); in BuildSwitchOnGeneratorState() local2885 NewSwitch(generator_state, in BuildSwitchOnGeneratorState()2935 Node* generator_state = in VisitSwitchOnGeneratorState() local2937 environment()->BindGeneratorState(generator_state); in VisitSwitchOnGeneratorState()