

from typing import *

# Someday TensorFlow will have types!
Tensor = Any

class CellContext(NamedTuple):
	features: Dict
	args: Dict
	vocab_embedding: Tensor
	in_prev_outputs: Tensor
	in_iter_id: Tensor
	in_iter_question_state: Tensor
	in_memory_state: Tensor
	in_question_tokens: Tensor
	in_question_tokens_padded: Tensor
	in_question_state: Tensor
	in_node_state: Tensor
	control_state: Tensor